hdu1520树形dp入门

题目链接

题意：要开派对，邀请了上司就不能邀请他的下属，邀请了下属就不能邀请他的上司，每个人有一个值，求邀请的人的总值最大

第一行给出一个数n，代表有n个人。

下面n行分别给出n个人的的值

再下面n行每行给出L,K;K是L的上司

以0 0结束一组输入

树形dp:把每个人看成一个点，则该点有两个状态：邀请或没被邀请

定义f[u][0]为节点没被邀请时的值；f[u][1]为节点被邀请时的值

状态转移方程：

f[u][0]=sum(max(f[v][0],f[v][1])//v为u的下属

f[u][1]=sum(f[v][0])//上司邀请了，则下属只有没被邀请这种状态

要求总值，则显然要求max(f[U][0],f[U][1])//U为根节点

用dfs遍历树用

f[u][0]=0;
f[u][1]=value[u];
vis[u]=1;

在递归中初始化每个节点的状态

达到实现动态规划的目的

代码实现：

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<queue>
 #include<vector>
 #include<algorithm>
 using namespace std;
 vector<int>vec[];//储存节点的子节点
 int value[];
 int vis[];
 int f[][];
 int in[];//入度找根节点：根节点入度为0
 void dfs(int u)
 {
     f[u][]=;
     f[u][]=value[u];
     vis[u]=;
     for(int i=;i<vec[u].size();i++)
     {
         int v=vec[u][i];
         if(vis[v])continue;
         dfs(v);
         f[u][]+=max(f[v][],f[v][]);
         f[u][]+=f[v][];
     }
     return;
 }
 int main()
 {
     int n,u,v,i,j,k;
     while(~scanf("%d",&n)&&n)
     {
         for(i=;i<=n;i++)vec[i].clear();
         for(i=;i<=n;i++)scanf("%d",&value[i]);
         memset(in,,sizeof(in));
         while(~scanf("%d%d",&v,&u)&&(u+v))
         {
             vec[u].push_back(v);
             in[v]++;
         }
         for(i=;i<=n;i++)
         {
             if(!in[i])
             {
                 memset(vis,,sizeof(vis));
                 memset(f,,sizeof(f));
                 dfs(i);
                 printf("%d\n",max(f[i][],f[i][]));
                 break;
             }
         }
     }
     return ;
 }