// Jakub Rozek
// Miny
// PA 2022 A r4
// czas: ?
// pami: ?
#include "bits/stdc++.h"
using namespace std;
const int N=100005;
long long n,x,y,z,nr,odp;
long long r[N];
vector<pair<int,long long>> tv[N];
int odw[N];
void dfs(int v, int o, long long w)
{
if(w<0) return;
if(!odw[v])
{
odw[v]=1;
dfs(v,v,r[v]);
}
for(auto i:tv[v])
{
if(i.first==o) continue;
dfs(i.first, v, w-i.second);
}
}
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);;
cin>>n;
for(int i=1; i<=n; ++i) cin>>r[i];
for(int i=1; i<n; ++i)
{
cin>>x>>y>>z;
tv[x].push_back({y,z});
tv[y].push_back({x,z});
}
for(int i=1; i<=n; ++i)
{
++nr;
odw[i]=1;
dfs(i,i,r[i]);
cout<<odp<<" ";
}
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 | // Jakub Rozek // Miny // PA 2022 A r4 // czas: ? // pami: ? #include "bits/stdc++.h" using namespace std; const int N=100005; long long n,x,y,z,nr,odp; long long r[N]; vector<pair<int,long long>> tv[N]; int odw[N]; void dfs(int v, int o, long long w) { if(w<0) return; if(!odw[v]) { odw[v]=1; dfs(v,v,r[v]); } for(auto i:tv[v]) { if(i.first==o) continue; dfs(i.first, v, w-i.second); } } int main() { ios_base::sync_with_stdio(0); cin.tie(0);; cin>>n; for(int i=1; i<=n; ++i) cin>>r[i]; for(int i=1; i<n; ++i) { cin>>x>>y>>z; tv[x].push_back({y,z}); tv[y].push_back({x,z}); } for(int i=1; i<=n; ++i) { ++nr; odw[i]=1; dfs(i,i,r[i]); cout<<odp<<" "; } return 0; } |
English