蓝桥杯 生命之树 java_[蓝桥杯2015初赛]生命之树 （dfs）

[蓝桥杯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)相关教程