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