最大子树和——树形dp

P1122 最大子树和 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

一道很好的树形dp基础题。但是自己因为刚开始刷的题被限制了思路。。。。其实这道题很简单。

首先是无向图存储。

dfs遍历，如果是子节点，并且子节点的总值大于0，那就加上，不然不要。

#include <bits/stdc++.h>
using namespace std;
const int N=16200;
int ans=-2147483647;
int a[N],f[N],n;
vector<int> e[N];

void dfs(int u,int fa)
{
    f[u]=a[u];
    for(int i=0;i<e[u].size();i++)
    {
        int y=e[u][i];
        if(y!=fa)
        {
            dfs(y,u);
            if(f[y]>0)f[u]+=f[y];
        }
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    
    int x,y;
    for(int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&x,&y);
        e[x].push_back(y);
        e[y].push_back(x);
    }
    
    dfs(1,0);
    for(int i=1;i<=n;i++)ans=max(ans,f[i]);
    printf("%d\n",ans);
    
    return 0;
}