2366 砍树（树形dp）

51nod 2366
题解：我们定义dp[i][0]表示以i为根节点的子树白色点比黑色点最多的个数（可能会出现负数，具体实现看代码，dp[i][1]同理）

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=1e5+10;
vector<int> G[maxn];
int c[maxn];
int n;
int dp[maxn][2];
int res;
void dfs(int u,int fa)
{
    int len=G[u].size();
    int s1=0,s0=0;
    for(int i=0;i<len;i++){
        int to=G[u][i];
        if(to==fa)continue;
        dfs(to,u);
        if(dp[to][1]>0){
        s1+=dp[to][1];
        }
        if(dp[to][0]>0){
        s0+=dp[to][0];
        }
    }
    if(c[u]==0){
        dp[u][0]=s0+1;
        dp[u][1]=s1-1;
        dp[u][0]=max(dp[u][0],1);
    }
    else{
        dp[u][0]=s0-1;
        dp[u][1]=s1+1;
        dp[u][1]=max(dp[u][1],1);
    }
    //printf("dp[%d][0]:%d dp[%d][1]:%d\n",u,dp[u][0],u,dp[u][1]);
    res=max(res,max(dp[u][0],dp[u][1]));
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&c[i]);
    }
    int u,v;
    for(int i=1;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    res=0;
    dfs(1,0);
    printf("%d\n",res);
    return 0;
}