最大子树和模板

最新推荐文章于 2023-03-11 08:54:52 发布

剑气扬名

最新推荐文章于 2023-03-11 08:54:52 发布

阅读量179

点赞数

本文链接：https://blog.csdn.net/HelloACM_ICPC/article/details/104640303

版权

最大子树和

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

void add(int x,int y) 
{
    tree[++cnt].next=head[x];
    tree[cnt].to=y;
    head[x]=cnt;
}
void dfs(int u,int fa)
{
    f[u]=a[u]; 
    for(int i=head[u];i;i=tree[i].next) 
        int v=tree[i].to;  
        if(v!=fa)  
        {
            dfs(v,u); 
            f[u]+=max(0,f[v]);  
        }
    }
    ans=max(ans,f[u]);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x); 
    }
    dfs(1,0);  
    printf("%d\n",ans);
}