[P1352]沒有上司的舞會

原題鏈接

樹形DP入門
DP方程搞錯了居然還過了90
(*/ω＼*)

利用DFS遞歸求解
每個點分爲選和不選兩種情況
假設選爲1不選爲0

        dp[x][0]+=max(dp[num[i]][1],dp[num[i]][0]);
        //這裏一開始寫成了dp[num[i]][1] 但它的兒子的兩種狀態實際上都是可選的

        dp[x][1]+=dp[num[i]][0];

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<climits>
#include<queue>
#define LL long long
using namespace std;

int cnt,num[12005],nxt[12005],head[6005],dp[6005][2],happy[6005],n,f[6005],boss,ans;

void add(int p1,int p2)
{
    cnt++;
    num[cnt]=p2;
    nxt[cnt]=head[p1];
    head[p1]=cnt;
}

void dfs(int x)
{
    int i;

    dp[x][0]=0;
    dp[x][1]=happy[x];

    for(i=head[x];i;i=nxt[i])
    {
        dfs(num[i]);
        dp[x][0]+=max(dp[num[i]][1],dp[num[i]][0]);//!!!!
        dp[x][1]+=dp[num[i]][0];
    }

    return;
}

int main()
{
    int i,l,k;

    scanf("%d",&n);

    for(i=1;i<=n;i++)
        scanf("%d",&happy[i]);

    for(i=1;i<n;i++)        
    {
        scanf("%d%d",&l,&k);
        add(k,l);
        f[l]=k;
    }

    for(i=1;i<=n;i++)
        if(!f[i])
        {
            boss=i;
            break;
        }

    dfs(boss);
    for(i=1;i<=n;i++)
        ans=max(ans,max(dp[i][0],dp[i][1]));

    printf("%d",ans);
    return 0;
}