#include <bits/stdc++.h> #define FOR(i,a,b) for(int i=(a);i<(b);++i) #define FORD(i, a, b) for(int i = (a); i >= (b); --i) #define VAR(v, i) __typeof(i) v=(i) #define FORE(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define all(v) (v).begin(),(v).end() #define PII pair<int,int> #define mp make_pair #define st first #define nd second #define pb push_back #define lint long long int #define VI vector<int> #define debug(x) {cerr <<#x <<" = " <<x <<endl; } #define debug2(x,y) {cerr <<#x <<" = " <<x << ", "<<#y<<" = "<< y <<endl; } #define debug3(x,y,z) {cerr <<#x <<" = " <<x << ", "<<#y<<" = "<< y << ", " << #z << " = " << z <<endl; } #define debugv(x) {{cerr <<#x <<" = "; FORE(itt, (x)) cerr <<*itt <<", "; cerr <<endl; }} #define debugt(t,n) {{cerr <<#t <<" = "; FOR(it,0,(n)) cerr <<t[it] <<", "; cerr <<endl; }} #define make( x) int (x); scanf("%d",&(x)); #define make2( x, y) int (x), (y); scanf("%d%d",&(x),&(y)); #define make3(x, y, z) int (x), (y), (z); scanf("%d%d%d",&(x),&(y),&(z)); #define make4(x, y, z, t) int (x), (y), (z), (t); scanf("%d%d%d%d",&(x),&(y),&(z),&(t)); #define makev(v,n) VI (v); FOR(i,0,(n)) { make(a); (v).pb(a);} #define IOS ios_base::sync_with_stdio(0) #define HEAP priority_queue #define read( x) scanf("%d",&(x)); #define read2( x, y) scanf("%d%d",&(x),&(y)); #define read3(x, y, z) scanf("%d%d%d",&(x),&(y),&(z)); #define read4(x, y, z, t) scanf("%d%d%d%d",&(x),&(y),&(z),&(t)); #define readv(v,n) FOR(i,0,(n)) { make(a); (v).pb(a);} using namespace std; const int max_n = 100005; int n, q; lint zisk[max_n]; VI g[max_n]; vector<lint> strata[max_n]; int xi, yi; lint ci; lint huhu[max_n]; void dfs(int v, int p) { FOR(i,0,g[v].size()) { int w = g[v][i]; if (v == xi && w == yi) strata[v][i] = ci; if (w == p) continue; huhu[w] = huhu[v] + strata[v][i]; dfs(w, v); } } int licz(int x) { FOR(i,0,n) huhu[i] = 0; dfs(x,-1); int kto = (x == 0) ? 1 : 0; FOR(i,0,n) { if (i!=x && zisk[i]-huhu[i] > zisk[kto]-huhu[kto]) { kto = i; } } return kto; } int main() { read2(n, q); FOR(i,0,n) { lint a; scanf("%lld", &a); zisk[i] = a; } FOR(i,0,n-1) { int a, b; lint c; scanf("%d%d%lld",&a,&b,&c); a--; b--; g[a].pb(b); g[b].pb(a); strata[a].pb(c); strata[b].pb(c); } int gdzie = 0; while (q--) { make(type); if (type == 1) { make(v); v--; lint d; scanf("%lld", &d); zisk[v] = d; } else { int a, b; lint c; scanf("%d%d%lld",&a, &b, &c); a--; b--; xi = a, yi = b, ci = c; } gdzie = licz(gdzie); printf("%d ", gdzie + 1); } printf("\n"); }
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 | #include <bits/stdc++.h> #define FOR(i,a,b) for(int i=(a);i<(b);++i) #define FORD(i, a, b) for(int i = (a); i >= (b); --i) #define VAR(v, i) __typeof(i) v=(i) #define FORE(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define all(v) (v).begin(),(v).end() #define PII pair<int,int> #define mp make_pair #define st first #define nd second #define pb push_back #define lint long long int #define VI vector<int> #define debug(x) {cerr <<#x <<" = " <<x <<endl; } #define debug2(x,y) {cerr <<#x <<" = " <<x << ", "<<#y<<" = "<< y <<endl; } #define debug3(x,y,z) {cerr <<#x <<" = " <<x << ", "<<#y<<" = "<< y << ", " << #z << " = " << z <<endl; } #define debugv(x) {{cerr <<#x <<" = "; FORE(itt, (x)) cerr <<*itt <<", "; cerr <<endl; }} #define debugt(t,n) {{cerr <<#t <<" = "; FOR(it,0,(n)) cerr <<t[it] <<", "; cerr <<endl; }} #define make( x) int (x); scanf("%d",&(x)); #define make2( x, y) int (x), (y); scanf("%d%d",&(x),&(y)); #define make3(x, y, z) int (x), (y), (z); scanf("%d%d%d",&(x),&(y),&(z)); #define make4(x, y, z, t) int (x), (y), (z), (t); scanf("%d%d%d%d",&(x),&(y),&(z),&(t)); #define makev(v,n) VI (v); FOR(i,0,(n)) { make(a); (v).pb(a);} #define IOS ios_base::sync_with_stdio(0) #define HEAP priority_queue #define read( x) scanf("%d",&(x)); #define read2( x, y) scanf("%d%d",&(x),&(y)); #define read3(x, y, z) scanf("%d%d%d",&(x),&(y),&(z)); #define read4(x, y, z, t) scanf("%d%d%d%d",&(x),&(y),&(z),&(t)); #define readv(v,n) FOR(i,0,(n)) { make(a); (v).pb(a);} using namespace std; const int max_n = 100005; int n, q; lint zisk[max_n]; VI g[max_n]; vector<lint> strata[max_n]; int xi, yi; lint ci; lint huhu[max_n]; void dfs(int v, int p) { FOR(i,0,g[v].size()) { int w = g[v][i]; if (v == xi && w == yi) strata[v][i] = ci; if (w == p) continue; huhu[w] = huhu[v] + strata[v][i]; dfs(w, v); } } int licz(int x) { FOR(i,0,n) huhu[i] = 0; dfs(x,-1); int kto = (x == 0) ? 1 : 0; FOR(i,0,n) { if (i!=x && zisk[i]-huhu[i] > zisk[kto]-huhu[kto]) { kto = i; } } return kto; } int main() { read2(n, q); FOR(i,0,n) { lint a; scanf("%lld", &a); zisk[i] = a; } FOR(i,0,n-1) { int a, b; lint c; scanf("%d%d%lld",&a,&b,&c); a--; b--; g[a].pb(b); g[b].pb(a); strata[a].pb(c); strata[b].pb(c); } int gdzie = 0; while (q--) { make(type); if (type == 1) { make(v); v--; lint d; scanf("%lld", &d); zisk[v] = d; } else { int a, b; lint c; scanf("%d%d%lld",&a, &b, &c); a--; b--; xi = a, yi = b, ci = c; } gdzie = licz(gdzie); printf("%d ", gdzie + 1); } printf("\n"); } |