hdu1520 树形dp

Posted 2020-08-15 a_clown_cz

题目链接：http://acm.split.hdu.edu.cn/showproblem.php?pid=1520
题意：上司和直系下属不能同时参加party，求party的最大活跃值。
输入： 输入n个结点，接下来的n行，表示1~n的每个结点分别具有的活跃值，输入a,b，表示b是a的上司，当a==0&& b==0时，该组数据输入结束。
输出：由于直接有上司和下属关系的两个人不能同时参加party， 求出能让party活跃值最大的方案（求出最大的活跃值即可）.
状态转移方程： dp[i][1]+=dp[j][0]； 　　　　　　　 

　　　　　　　 dp[i][0]+=max(dp[j][0],dp[j][0]);

其中i是j的父亲结点。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int N=6005;

struct node
{
    int v;
    int next;
} node[N<<1];

int dp[N][2];
int a[N],head[N],in[N];
int k;

void addedge(int u,int v)  ///u是父亲，v是孩子
{
    node[k].v=v;
    node[k].next=head[u];
    head[u]=k++;
}

int dfs(int u)
{
    dp[u][0]=0;
    dp[u][1]=a[u];
    for(int i=head[u]; i!=-1; i=node[i].next)
    {
        int v=node[i].v;
        dfs(v);
        dp[u][0]+=max(dp[v][1],dp[v][0]);
        dp[u][1]+=dp[v][0];
    }
    return max(dp[u][1],dp[u][0]);
}

int main()
{
    int n,x,y;
    while(scanf("%d",&n)==1)
    {
        k=0;
        memset(head,-1,sizeof(head));
        memset(dp,0,sizeof(dp));
        memset(in,0,sizeof(in));
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        while(scanf("%d%d",&x,&y) && (x+y))
        {
            addedge(y,x);
            in[x]++;
        }
        int ans=0;
        for(int i=1; i<=n; i++)  ///其实这是一个森林，找到每棵树的根，求和
            if(!in[i]) ans+=dfs(i);
        printf("%d\n",ans);
    }
    return 0;
}