[蓝桥杯2015初赛]生命之树 (dfs)
[蓝桥杯2015初赛]生命之树 (dfs)
代码:
#include
#include
#include
#include
#include
#include
using namespace std;
#define ll long long
const int maxn=1e5+5;
#include
int q[maxn]; ///每个点的权重
ll qq[maxn]; ///每个点作为根节点时能得到的最大权和
vectorv[maxn]; ///邻接表
ll ans;
void dfs(int root,int fa) ///以root为根,算出最大权和
{
qq[root]=(ll)q[root];
for(int i=0;i
{
int son=v[root][i]; ///root的一个孩子
if(son!=fa)
{
dfs(son,root);
if(qq[son]>0)
{
qq[root]+=qq[son];
}
}
}
if(qq[root]>ans)
{
ans=qq[root];
}
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&q[i]);
}
for(int i=1;i<=n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
v[x].push_back(y);
v[y].push_back(x);
}
ans=0;
dfs(1,0);
printf("%lld\n",ans);
return 0;
}
[蓝桥杯2015初赛]生命之树 (dfs)相关教程