#include<bits/stdc++.h>
using namespace std;
int n,k,m;
long long infi = (long long)1e18 + 7;
vector<pair<int,int>> zelki[7010];
long long koszt[7010][7010]; //koszt[i][j] - cena za paczke zelkow o wadze j oraz do ktorej uzyto i pierwszych kolorow zelkow
vector<pair<long long, int>> nominaly; //nominaly[i] = {a,b} - mamy kombinacje zelkow dobrokolorowych o masie b i koszcie a
long long best[7010]; // best[i] - najlepsza cena za paczke o wadze i
int main(){
cin.tie(0);
cout.tie(0);
ios_base::sync_with_stdio(false);
cin >> n >> k >> m;
int ki,mi,ci;
for(int i = 0; i < n; i++){
cin >> ki >> mi >> ci;
zelki[ki].push_back({mi,ci});
}
for(int i = 0; i <= k; i++){
for(int j = 0; j <= m; j++){
koszt[i][j] = infi;
}
}
koszt[0][0] = 0;
int tmp;
for(int i = 1; i <= k; i++){
for(int j = 0; j < m; j++){
if(koszt[i-1][j] == infi) continue;
for(auto zelek : zelki[i]){
tmp = (j + zelek.first) % m;
koszt[i][tmp] = min(koszt[i][tmp], koszt[i-1][j] + zelek.second);
}
}
}
// for(int i = 0; i <= k; i++){
// for(int j = 0; j <= m; j++){
// cout << koszt[i][j] << " ";
// }
// cout << endl;
// }
// cout << endl;
priority_queue<pair<long long, int>, vector<pair<long long, int>>, greater<pair<long long, int>>> pque;
for(int i = 0; i < m; i++){
best[i] = koszt[k][i];
if(koszt[k][i] != infi){
// nominaly.push_back({i,koszt[k][i]});
pque.push({koszt[k][i],i});
}
}
pair<long long, int> tmpp;
int ctr1 = 0;
best[0] = 0;
while(ctr1 < m - 1 && !pque.empty()){
tmpp = pque.top();
pque.pop();
if(tmpp.first > best[tmpp.second]) continue;
ctr1++;
nominaly.push_back(tmpp);
for(auto i : nominaly){
tmp = (tmpp.second + i.second) % m;
if(best[tmp] > tmpp.first + i.first){
best[tmp] = tmpp.first + i.first;
pque.push({best[tmp], tmp});
}
}
}
for(int i = 0; i < m; i++){
if(best[i] == infi) best[i] = -1;
cout << best[i] << "\n";
}
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 | #include<bits/stdc++.h> using namespace std; int n,k,m; long long infi = (long long)1e18 + 7; vector<pair<int,int>> zelki[7010]; long long koszt[7010][7010]; //koszt[i][j] - cena za paczke zelkow o wadze j oraz do ktorej uzyto i pierwszych kolorow zelkow vector<pair<long long, int>> nominaly; //nominaly[i] = {a,b} - mamy kombinacje zelkow dobrokolorowych o masie b i koszcie a long long best[7010]; // best[i] - najlepsza cena za paczke o wadze i int main(){ cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(false); cin >> n >> k >> m; int ki,mi,ci; for(int i = 0; i < n; i++){ cin >> ki >> mi >> ci; zelki[ki].push_back({mi,ci}); } for(int i = 0; i <= k; i++){ for(int j = 0; j <= m; j++){ koszt[i][j] = infi; } } koszt[0][0] = 0; int tmp; for(int i = 1; i <= k; i++){ for(int j = 0; j < m; j++){ if(koszt[i-1][j] == infi) continue; for(auto zelek : zelki[i]){ tmp = (j + zelek.first) % m; koszt[i][tmp] = min(koszt[i][tmp], koszt[i-1][j] + zelek.second); } } } // for(int i = 0; i <= k; i++){ // for(int j = 0; j <= m; j++){ // cout << koszt[i][j] << " "; // } // cout << endl; // } // cout << endl; priority_queue<pair<long long, int>, vector<pair<long long, int>>, greater<pair<long long, int>>> pque; for(int i = 0; i < m; i++){ best[i] = koszt[k][i]; if(koszt[k][i] != infi){ // nominaly.push_back({i,koszt[k][i]}); pque.push({koszt[k][i],i}); } } pair<long long, int> tmpp; int ctr1 = 0; best[0] = 0; while(ctr1 < m - 1 && !pque.empty()){ tmpp = pque.top(); pque.pop(); if(tmpp.first > best[tmpp.second]) continue; ctr1++; nominaly.push_back(tmpp); for(auto i : nominaly){ tmp = (tmpp.second + i.second) % m; if(best[tmp] > tmpp.first + i.first){ best[tmp] = tmpp.first + i.first; pque.push({best[tmp], tmp}); } } } for(int i = 0; i < m; i++){ if(best[i] == infi) best[i] = -1; cout << best[i] << "\n"; } return 0; } |
English