#include <bits/stdc++.h> using namespace std; int n, q, akt; long long wynik, wyg; long long z[100001]; vector<int> pol[100001]; vector<long long> kos[100001]; bool odw[100001]; void reset() { for(int i = 1; i <= n; i++) odw[i]=0; } void DFS(int x, int ko) { odw[x]=1; wynik -= ko; if(akt!=x && wynik+z[x]==wyg && x<akt) akt = x; if(akt!=x && wynik+z[x]>wyg) { wyg = wynik+z[x]; akt = x; } for(int i = 0; i < pol[x].size(); i++) if(!odw[pol[x][i]]) { DFS(pol[x][i], kos[x][i]); } wynik += ko; } int main() { ios::sync_with_stdio(0); cin.tie(0); cin >> n >> q; for(int i = 1; i <= n; i++) cin >> z[i]; int a, b, v, d; long long c; for(int i = 0; i < n-1; i++) { cin >> a >> b >> d; pol[a].push_back(b); kos[a].push_back(d); pol[b].push_back(a); kos[b].push_back(d); } int co; akt = 1; for(int g = 0; g < q; g++) { cin >> co; if(co==1) { cin >> v >> d; z[v]=d; } else { cin >> a >> b >> d; for(int i = 0; i < pol[a].size(); i++) if(pol[a][i]==b) kos[a][i]=d; for(int i = 0; i < pol[b].size(); i++) if(pol[b][i]==a) kos[b][i]=d; wynik = 0; } wynik = 0; wyg = -9000000000000000000; DFS(akt, 0); cout << akt << " "; reset(); } 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 | #include <bits/stdc++.h> using namespace std; int n, q, akt; long long wynik, wyg; long long z[100001]; vector<int> pol[100001]; vector<long long> kos[100001]; bool odw[100001]; void reset() { for(int i = 1; i <= n; i++) odw[i]=0; } void DFS(int x, int ko) { odw[x]=1; wynik -= ko; if(akt!=x && wynik+z[x]==wyg && x<akt) akt = x; if(akt!=x && wynik+z[x]>wyg) { wyg = wynik+z[x]; akt = x; } for(int i = 0; i < pol[x].size(); i++) if(!odw[pol[x][i]]) { DFS(pol[x][i], kos[x][i]); } wynik += ko; } int main() { ios::sync_with_stdio(0); cin.tie(0); cin >> n >> q; for(int i = 1; i <= n; i++) cin >> z[i]; int a, b, v, d; long long c; for(int i = 0; i < n-1; i++) { cin >> a >> b >> d; pol[a].push_back(b); kos[a].push_back(d); pol[b].push_back(a); kos[b].push_back(d); } int co; akt = 1; for(int g = 0; g < q; g++) { cin >> co; if(co==1) { cin >> v >> d; z[v]=d; } else { cin >> a >> b >> d; for(int i = 0; i < pol[a].size(); i++) if(pol[a][i]==b) kos[a][i]=d; for(int i = 0; i < pol[b].size(); i++) if(pol[b][i]==a) kos[b][i]=d; wynik = 0; } wynik = 0; wyg = -9000000000000000000; DFS(akt, 0); cout << akt << " "; reset(); } return 0; } |