// 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; } |