#include <bits/stdc++.h>
#define MAX_N 100007
using namespace std;
long long mina[MAX_N];
vector<pair<long long, long long>> graf[MAX_N];
long long odw[MAX_N];
void dfs(long long w, long long k, long long p){
if(k < mina[w]){
k = mina[w];
p = w;
}
odw[w] = p;
for(auto& i : graf[w]){
if(odw[i.first] != p && i.second <= k) dfs(i.first, k-i.second, p);
}
}
int main(){
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
long long n, a, b, c;
cin >> n;
for(int i = 1; i <= n; i++) cin >> mina[i];
for(int i = 1; i < n; i++){
cin >> a >> b >> c;
graf[a].push_back({b, c});
graf[b].push_back({a, c});
}
long long wynik;
for(int i = 1; i <= n; i++){
dfs(i, mina[i], i);
wynik = 0;
for(int j = 1; j <= n; j++){
if(odw[j]){
wynik++;
odw[j] = 0;
}
}
cout << wynik << " ";
}
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 | #include <bits/stdc++.h> #define MAX_N 100007 using namespace std; long long mina[MAX_N]; vector<pair<long long, long long>> graf[MAX_N]; long long odw[MAX_N]; void dfs(long long w, long long k, long long p){ if(k < mina[w]){ k = mina[w]; p = w; } odw[w] = p; for(auto& i : graf[w]){ if(odw[i.first] != p && i.second <= k) dfs(i.first, k-i.second, p); } } int main(){ ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); long long n, a, b, c; cin >> n; for(int i = 1; i <= n; i++) cin >> mina[i]; for(int i = 1; i < n; i++){ cin >> a >> b >> c; graf[a].push_back({b, c}); graf[b].push_back({a, c}); } long long wynik; for(int i = 1; i <= n; i++){ dfs(i, mina[i], i); wynik = 0; for(int j = 1; j <= n; j++){ if(odw[j]){ wynik++; odw[j] = 0; } } cout << wynik << " "; } return 0; } |
English