洛谷1352没有上司的舞会——树型dp

题目：https://www.luogu.org/problemnew/show/P1352

#include<iostream>
#include<cstdio>
using namespace std;
int n,a[],l[],cnt,x,y,f[][],sum;
struct Node{
    int to,next;
}edge[];
bool b[],rd[];
void add(int x,int y)
{
    cnt++;
    edge[cnt].next=l[x];
    edge[cnt].to=y;
    l[x]=cnt;
    rd[y]=;
}
void dfs(int cur)
{
    if(b[cur])return;
    for(int i=l[cur];i;i=edge[i].next)
    {
        int v=edge[i].to;
        dfs(v);
        f[cur][]+=max(f[v][],f[v][]);
        f[cur][]+=f[v][];
    }
    f[cur][]+=a[cur];
    b[cur]=;
}
int main()
{
    scanf("%d",&n);
    for(int i=;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=;i<=n;i++)
    {
        scanf("%d%d",&x,&y);
        add(y,x);
    }
    for(int i=;i<=n;i++)
    {
        dfs(i);
        if(!rd[i])sum=max(sum,sum+max(f[i][],f[i][]));
    }
    printf("%d",sum);
    return ;
}