1 Star 0 Fork 1

saigon/Algorithms

forked from charlieshu/Algorithms 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
P1122 ╫ю┤є╫╙╩ў║═.cpp 1.08 KB
一键复制 编辑 原始数据 按行查看 历史
charlie 提交于 2024-01-09 00:01 . move from github to gitee
#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;
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/saigonshu/algorithm.git
git@gitee.com:saigonshu/algorithm.git
saigonshu
algorithm
Algorithms
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385