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