代码拉取完成,页面将自动刷新
同步操作将从 charlieshu/Algorithms 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
#include <iostream>
#include <math.h>
#include <string.h>
#include <vector>
#include <queue>
using namespace std;
int main(){
int ans=0;
int n;
cin>>n;
int a[n];
for(int i=0;i<n;i++)
cin>>a[i];
vector<int> f[n];
int dp[n];
int edge[n];
memset(edge,0,sizeof(edge));
memset(dp,0,sizeof(dp));
for(int i=0;i<n-1;i++){
int s,t;
cin>>s>>t;
s--;
t--;
edge[s]++;
edge[t]++;
f[s].push_back(t);
f[t].push_back(s);
}
queue<int> q;
for(int i=0;i<n;i++)
if(edge[i] == 1)
q.push(i);
while(!q.empty()){
int now;
now = q.front();
if(edge[now] == 0){
ans = max(ans,dp[now]+a[now]);
break;
}
q.pop();
int index;
for(int i=0;i<f[now].size();i++){
if(edge[f[now][i]] > 1){
index = i;
break;
}
}
edge[f[now][index]]--;
if(edge[f[now][index]] == 1)
q.push(f[now][index]);
dp[now] += a[now];
ans = max(ans,dp[now]);
if(dp[now] <= 0)
continue;
dp[f[now][index]] += dp[now];
}
for(int i=0;i<n;i++)
cout<<dp[i]<<" ";
cout<<endl;
cout<<ans;
return 0;
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。