HDU 1520 Anniversary party(树形dp)

来源：互联网发布：php 清空文件编辑：程序博客网时间：2024/05/16 16:57

题目链接：点击打开链接

题意：由于有上司和下属直接关系的两个人不能同时参加party，求出party最大的活跃值

每个结点有两种状态，参加和不参加，用0表示不参加，1表示参加

dp[i][1]表示第i个参与者参加了，dp[i][0]表示第i个参与者没有参加。

状态转移方程：dp[u][0] += max (dp[v][0], dp[v][1]) :表示上司没参加，其员工可以参加可以不参加

dp[u][1] += dp[v][0] : 表示若上司参加了，其员工一定不会参加

#include <cstdio>#include <cstring>#include <iostream>#include <vector>using namespace std;int n,num[6005];int dp[6005][2];//dp[i][1]:以当前节点i为父节点的 子树，取i时的最大值  dp[i][0]:不取i时的最大值vector<int> v[6005];void DP(int x){dp[x][1]=num[x];dp[x][0]=0;for(int i=0;i<v[x].size();i++){DP(v[x][i]);dp[x][0]+=max(dp[v[x][i]][0],dp[v[x][i]][1]);dp[x][1]+=dp[v[x][i]][0];}}int main(){int root,x,y;int book[6005];while(scanf("%d",&n)!=EOF){memset(book,0,sizeof(book));for(int i=1;i<=n;i++){v[i].clear();}for(int i=1;i<=n;i++){scanf("%d",&num[i]);}while(scanf("%d%d",&x,&y)!=EOF&&x&&y){v[y].push_back(x);book[x]=1;}for(int i=1;i<=n;i++){//找树根if(book[i]==0){DP(root=i);break;}}printf("%d\n",max(dp[root][0],dp[root][1]));}return 0;}

1 0