刚看了下,感觉是对的,但是为什么我的全部RE。
#include<bits/stdc++.h>
#define MAXN 16005
using namespace std;
int n,a[MAXN],dp[MAXN];
vector<int> sx[MAXN];
int dfs(int now,int last){
dp[now]=a[now];
for(int i=0;i<sx[now].size();i++){
if(last!=sx[now][i]){
dfs(sx[now][i],now);
if(dp[sx[now][i]]>0){
dp[now]=dp[now]+dp[sx[now][i]];
}
}
}
}
int main(){
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
}
for(int i=1;i<=n;i++){
int u,v;
cin>>u>>v;
sx[u].push_back(v);
sx[v].push_back(u);
}
dfs(1,0);
int ans=-INT_MAX;
for(int i=1;i<=n;i++){
ans=max(ans,dp[i]);
}
cout<<ans;
return 0;
}