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