#include<bits/stdc++.h>
#define fi first
#define se second
using namespace std;
const int N=1e5;
vector<pair<int,long long>> e[N+10];
long long range[N+10];
long long mx[N+10];
int dfs(int x,long long r)
{
int ans=0;
if(mx[x]==-1)
ans++;
r=max(r,range[x]);
mx[x]=r;
for(auto [v,c]:e[x])
{
if(r-c>mx[v])
ans+=dfs(v,r-c);
}
return ans;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
int n;
cin>>n;
for(int i=1;i<=n;i++)
cin>>range[i];
for(int i=1;i<n;i++)
{
int a,b;
long long c;
cin>>a>>b>>c;
e[a].emplace_back(b,c);
e[b].emplace_back(a,c);
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
mx[j]=-1;
cout<<dfs(i,0)<<" ";
}
cout<<"\n";
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 | #include<bits/stdc++.h> #define fi first #define se second using namespace std; const int N=1e5; vector<pair<int,long long>> e[N+10]; long long range[N+10]; long long mx[N+10]; int dfs(int x,long long r) { int ans=0; if(mx[x]==-1) ans++; r=max(r,range[x]); mx[x]=r; for(auto [v,c]:e[x]) { if(r-c>mx[v]) ans+=dfs(v,r-c); } return ans; } int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); int n; cin>>n; for(int i=1;i<=n;i++) cin>>range[i]; for(int i=1;i<n;i++) { int a,b; long long c; cin>>a>>b>>c; e[a].emplace_back(b,c); e[b].emplace_back(a,c); } for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) mx[j]=-1; cout<<dfs(i,0)<<" "; } cout<<"\n"; return 0; } |
English