树形dp

今天开始做老师给的专辑，打开DP专辑 A题 Rebuilding Roads 直接不会了，发现是树形DP，百度了下了该题，看了老半天看不懂，想死的冲动都有了~~~~

最后百度了下，树形DP入门，找到了 poj 2342 Anniversary party   先入门一下~

题意：

某公司要举办一次晚会，但是为了使得晚会的气氛更加活跃，每个参加晚会的人都不希望在晚会中见到他的直接上司，现在已知每个人的活跃指数和上司关系（当然不可能存在环），求邀请哪些人（多少人）来能使得晚会的总活跃指数最大。

思路：

任何一个点的取舍可以看作一种决策，那么状态就是在某个点取的时候或者不取的时候，以他为根的子树能有的最大活跃总值。分别可以用f[i,1]和f[i,0]表示第i个人来和不来。

当i来的时候，dp[i][1] += dp[j][0];//j为i的下属

当i不来的时候，dp[i][0] +=max(dp[j][1],dp[j][0]);//j为i的下属

以下代码参考：http://hi.baidu.com/saintlleo/blog/item/0606b3feb7026ad3b48f3111.html

//AC CODE:

[cpp]  view plain copy print ?

1. #include<iostream>  
2. #include<cmath>  
3. #include<algorithm>  
4. #include<vector>  
5. #include<cstdio>  
6. #include<cstdlib>  
7. #include<cstring>  
8. #include<string>  
9.   
10. using namespace std;  
11.   
12. #define maxn 6005  
13.   
14. int n;  
15. int dp[maxn][2],father[maxn];//dp[i][0]0表示不去，dp[i][1]1表示去了  
16. bool visited[maxn];  
17.   
18. void tree_dp(int node)  
19. {  
20.     int i;  
21.     visited[node] = 1;  
22.     for(i=1; i<=n; i++)  
23.     {  
24.         if(!visited[i]&&father[i] == node)//i为下属  
25.         {  
26.             tree_dp(i);//递归调用孩子结点，从叶子结点开始dp  
27.             //关键  
28.             dp[node][1] += dp[i][0];//上司来,下属不来  
29.             dp[node][0] +=max(dp[i][1],dp[i][0]);//上司不来，下属来、不来  
30.         }  
31.     }  
32. }  
33.   
34. int main()  
35. {  
36.     int i;  
37.     int f,c,root;  
38.     while(scanf("%d",&n)!=EOF)  
39.     {  
40.         memset(dp,0,sizeof(dp));  
41.         memset(father,0,sizeof(father));  
42.         memset(visited,0,sizeof(visited));  
43.         for(i=1; i<=n; i++)  
44.         {  
45.             scanf("%d",&dp[i][1]);  
46.         }  
47.         root = 0;//记录父结点  
48.         bool beg = 1;  
49.         while (scanf("%d %d",&c,&f),c||f)  
50.         {  
51.             father[c] = f;  
52.             if( root == c || beg )  
53.             {  
54.                 root = f;  
55.             }  
56.         }  
57.         while(father[root])//查找父结点  
58.             root=father[root];  
59.         tree_dp(root);  
60.         int imax=max(dp[root][0],dp[root][1]);  
61.         printf("%d\n",imax);  
62.     }  
63.     return 0;  
64.   
65. }