#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; } |