#include<cstdio> #include<algorithm> #include<vector> using namespace std; #define MAX_N 7005 #ifndef DEB_VAL #define DEB_VAL 0 #endif #define DEB if(debug) #define MP make_pair #define PB push_back #define FT first #define SD second int debug = DEB_VAL; int n,k,m; vector< pair <int, pair<int, long long> > > in; int ink, inm; int reszta; long long wynik; long long inc; long long res[MAX_N]; long long tym[MAX_N]; int used[MAX_N]; int usedq[MAX_N]; vector< pair<int,long long> > qu[2]; vector< pair<int,long long> > quzel[2]; int main() { scanf("%d %d %d", &n, &k, &m); for(int i=0;i<n;i++) { scanf("%d %d %lld", &ink, &inm, &inc); in.PB(MP(ink,MP(inm,inc))); } sort(in.begin(),in.end()); for(int i=0;i<m;i++) { res[i]=-1; used[i]=-1; } res[0]=0; int akt=0; qu[akt].PB(MP(0,0)); int curk=-1; int ilekol=0; bool nowy=0; int uqiter=0; int aktzel=0; ilekol=0; for(int i=1;i<=m+1;i++) { DEB { printf("Iteracja: %d\n", i); } curk=-1; aktzel=0; quzel[aktzel].clear(); quzel[aktzel^1].clear(); quzel[aktzel].PB(MP(0,0)); if(i==1) { //przelatujemy wszystkie zelki i aktualizujemy reszty for(int j=0;j<n;j++) { DEB { printf("nowy zelek: %d -> %d %d %lld\n", j, in[j].FT, in[j].SD.FT, in[j].SD.SD); } if(in[j].FT!=curk) { if(curk!=-1) { DEB { printf("Wrzucam aktualne reszty - jest ich %d\n",uqiter); } //wrzucamy do quzel[akt] for(int x=0;x<uqiter;x++) { quzel[aktzel].PB(MP(usedq[x],tym[usedq[x]])); } } aktzel^=1; curk=in[j].FT; ilekol++; quzel[aktzel].clear(); uqiter=0; DEB { printf("Zmiana koloru! -> %d\n", curk); } } DEB { printf("Aktualne reszty:\n"); for(int x=0;x<quzel[aktzel^1].size();x++) { printf("%d %lld\n",quzel[aktzel^1][x].FT,quzel[aktzel^1][x].SD); } } //wrzucamy nowe na bazie starych for(int x=0;x<quzel[aktzel^1].size();x++) { reszta=(quzel[aktzel^1][x].FT+in[j].SD.FT)%m; DEB { printf("Nowe: %d %lld\n",reszta,quzel[aktzel^1][x].SD+in[j].SD.SD); } if(used[reszta]<i*10000+curk) { DEB { printf("Not used\n"); } used[reszta]=i*10000+curk; tym[reszta]=quzel[aktzel^1][x].SD+in[j].SD.SD; usedq[uqiter++]=reszta; } if(tym[reszta]>quzel[aktzel^1][x].SD+in[j].SD.SD) { tym[reszta]=quzel[aktzel^1][x].SD+in[j].SD.SD; } //TODO - to na koniec //qu[akt].PB(MP(reszta,qu[akt^1][x].SD+in[j].SD.SD)); } } DEB { printf("Koniec zelkow==========\n"); } } //uzupełniamy tablicę res i zostawiamy tylko te które coś zmieniły if(ilekol==k) { qu[akt^1].clear(); nowy=0; for(int x=0;x<qu[akt].size();x++) { DEB { printf("Stara reszta: %d %lld\n", qu[akt][x].FT, qu[akt][x].SD); } for(int y=0;y<uqiter;y++) { DEB { printf("Potencjalna dokładka: %d %lld\n",usedq[y],tym[usedq[y]]); } reszta=(qu[akt][x].FT+usedq[y])%m; wynik=qu[akt][x].SD+tym[usedq[y]]; if(res[reszta]==-1 || res[reszta]>wynik) { res[reszta]=wynik; qu[akt^1].PB(MP(reszta,wynik)); //TODO - tutaj moze byc też sporo nowy=1; } } } akt^=1; DEB { for(int x=0;x<qu[akt].size();x++) { printf("Mam: %d %lld\n",qu[akt][x].FT,qu[akt][x].SD); } } } else { break; } if(nowy==0) break; } for(int i=0;i<m;i++) { printf("%lld\n", res[i]); } return 0; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 | #include<cstdio> #include<algorithm> #include<vector> using namespace std; #define MAX_N 7005 #ifndef DEB_VAL #define DEB_VAL 0 #endif #define DEB if(debug) #define MP make_pair #define PB push_back #define FT first #define SD second int debug = DEB_VAL; int n,k,m; vector< pair <int, pair<int, long long> > > in; int ink, inm; int reszta; long long wynik; long long inc; long long res[MAX_N]; long long tym[MAX_N]; int used[MAX_N]; int usedq[MAX_N]; vector< pair<int,long long> > qu[2]; vector< pair<int,long long> > quzel[2]; int main() { scanf("%d %d %d", &n, &k, &m); for(int i=0;i<n;i++) { scanf("%d %d %lld", &ink, &inm, &inc); in.PB(MP(ink,MP(inm,inc))); } sort(in.begin(),in.end()); for(int i=0;i<m;i++) { res[i]=-1; used[i]=-1; } res[0]=0; int akt=0; qu[akt].PB(MP(0,0)); int curk=-1; int ilekol=0; bool nowy=0; int uqiter=0; int aktzel=0; ilekol=0; for(int i=1;i<=m+1;i++) { DEB { printf("Iteracja: %d\n", i); } curk=-1; aktzel=0; quzel[aktzel].clear(); quzel[aktzel^1].clear(); quzel[aktzel].PB(MP(0,0)); if(i==1) { //przelatujemy wszystkie zelki i aktualizujemy reszty for(int j=0;j<n;j++) { DEB { printf("nowy zelek: %d -> %d %d %lld\n", j, in[j].FT, in[j].SD.FT, in[j].SD.SD); } if(in[j].FT!=curk) { if(curk!=-1) { DEB { printf("Wrzucam aktualne reszty - jest ich %d\n",uqiter); } //wrzucamy do quzel[akt] for(int x=0;x<uqiter;x++) { quzel[aktzel].PB(MP(usedq[x],tym[usedq[x]])); } } aktzel^=1; curk=in[j].FT; ilekol++; quzel[aktzel].clear(); uqiter=0; DEB { printf("Zmiana koloru! -> %d\n", curk); } } DEB { printf("Aktualne reszty:\n"); for(int x=0;x<quzel[aktzel^1].size();x++) { printf("%d %lld\n",quzel[aktzel^1][x].FT,quzel[aktzel^1][x].SD); } } //wrzucamy nowe na bazie starych for(int x=0;x<quzel[aktzel^1].size();x++) { reszta=(quzel[aktzel^1][x].FT+in[j].SD.FT)%m; DEB { printf("Nowe: %d %lld\n",reszta,quzel[aktzel^1][x].SD+in[j].SD.SD); } if(used[reszta]<i*10000+curk) { DEB { printf("Not used\n"); } used[reszta]=i*10000+curk; tym[reszta]=quzel[aktzel^1][x].SD+in[j].SD.SD; usedq[uqiter++]=reszta; } if(tym[reszta]>quzel[aktzel^1][x].SD+in[j].SD.SD) { tym[reszta]=quzel[aktzel^1][x].SD+in[j].SD.SD; } //TODO - to na koniec //qu[akt].PB(MP(reszta,qu[akt^1][x].SD+in[j].SD.SD)); } } DEB { printf("Koniec zelkow==========\n"); } } //uzupełniamy tablicę res i zostawiamy tylko te które coś zmieniły if(ilekol==k) { qu[akt^1].clear(); nowy=0; for(int x=0;x<qu[akt].size();x++) { DEB { printf("Stara reszta: %d %lld\n", qu[akt][x].FT, qu[akt][x].SD); } for(int y=0;y<uqiter;y++) { DEB { printf("Potencjalna dokładka: %d %lld\n",usedq[y],tym[usedq[y]]); } reszta=(qu[akt][x].FT+usedq[y])%m; wynik=qu[akt][x].SD+tym[usedq[y]]; if(res[reszta]==-1 || res[reszta]>wynik) { res[reszta]=wynik; qu[akt^1].PB(MP(reszta,wynik)); //TODO - tutaj moze byc też sporo nowy=1; } } } akt^=1; DEB { for(int x=0;x<qu[akt].size();x++) { printf("Mam: %d %lld\n",qu[akt][x].FT,qu[akt][x].SD); } } } else { break; } if(nowy==0) break; } for(int i=0;i<m;i++) { printf("%lld\n", res[i]); } return 0; } |