#include <iostream> #include <vector> using namespace std; const int MAX = 200001; long long koszty[MAX], odw[MAX], ceny[MAX], iteracja = 1; vector < pair < int, int > > graf[MAX]; void DFS(int zrodlo, int poprz) { for (int i = 0; i < graf[zrodlo].size(); ++i) { if (graf[zrodlo][i].first != poprz) { koszty[graf[zrodlo][i].first] = koszty[zrodlo] + graf[zrodlo][i].second; DFS(graf[zrodlo][i].first, zrodlo); } } } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); long long a, b, c, d; int V, Q, zrodlo = 1; cin >> V >> Q; for (int i = 1; i <= V; ++i) { cin >> ceny[i]; } for (int i = 1; i < V; ++i) { cin >> a >> b >> c; graf[a].push_back({ b, c }); graf[b].push_back({ a, c }); } while (Q--) { cin >> a >> b >> c; if (a & 1) { ceny[b] = c; } else { cin >> d; for (int i = 0; i < graf[b].size(); ++i) { if (graf[b][i].first == c) { graf[b][i].second = d; break; } } for (int i = 0; i < graf[c].size(); ++i) { if (graf[c][i].first == b) { graf[c][i].second = d; break; } } } koszty[zrodlo] = 0; long long zysk = -1000000000000000, wie; DFS(zrodlo, zrodlo); for (int i = 1; i <= V; ++i) { if (i != zrodlo) { if (zysk < -koszty[i] + ceny[i]) { zysk = -koszty[i] + ceny[i]; wie = i; } } } zrodlo = wie; cout << wie << " "; } 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 | #include <iostream> #include <vector> using namespace std; const int MAX = 200001; long long koszty[MAX], odw[MAX], ceny[MAX], iteracja = 1; vector < pair < int, int > > graf[MAX]; void DFS(int zrodlo, int poprz) { for (int i = 0; i < graf[zrodlo].size(); ++i) { if (graf[zrodlo][i].first != poprz) { koszty[graf[zrodlo][i].first] = koszty[zrodlo] + graf[zrodlo][i].second; DFS(graf[zrodlo][i].first, zrodlo); } } } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); long long a, b, c, d; int V, Q, zrodlo = 1; cin >> V >> Q; for (int i = 1; i <= V; ++i) { cin >> ceny[i]; } for (int i = 1; i < V; ++i) { cin >> a >> b >> c; graf[a].push_back({ b, c }); graf[b].push_back({ a, c }); } while (Q--) { cin >> a >> b >> c; if (a & 1) { ceny[b] = c; } else { cin >> d; for (int i = 0; i < graf[b].size(); ++i) { if (graf[b][i].first == c) { graf[b][i].second = d; break; } } for (int i = 0; i < graf[c].size(); ++i) { if (graf[c][i].first == b) { graf[c][i].second = d; break; } } } koszty[zrodlo] = 0; long long zysk = -1000000000000000, wie; DFS(zrodlo, zrodlo); for (int i = 1; i <= V; ++i) { if (i != zrodlo) { if (zysk < -koszty[i] + ceny[i]) { zysk = -koszty[i] + ceny[i]; wie = i; } } } zrodlo = wie; cout << wie << " "; } return 0; } |