FZU_1615 Hero's RP

类似一个高中的DP题。树状DP
http://acm.fzu.edu.cn/problem.php?pid=1615
题目的意思就是邀请一些人，可以获得对应的RP价值，可是如果邀请到了有上下属关系的人，那么RP价值会减少，求的就是如何邀请才能使获得的RP最大
开个数组dp[i][2]表示第i个节点选择与否的解，然后从根DFS，最后线性扫描数组取max(dp[i][0],dp[i][1])(0<=i<n);
一开始就只以为根节点最大
WA了几十次，囧~
主要方程:
dp[i][0]=max(dp[i][0]+dp[j][0],dp[i][0]+dp[j][1]);//父节点不去，那么子节点取不取的情况都可以不用考虑RP减少问题
dp[i][1]=max(dp[i][1]+dp[j][0],dp[i][1]+dp[j][1]-(i,j)所减少的RP);//父节点去，那么就要考虑了

1. #include <iostream>
2. #include <string>
3. #include <map>
4. #include <vector>
5. using namespace std;
6. struct RP_node
7. {
8.     char name[11];
9.     int rp;
10.     vector<int> son;
11.     vector<int> sub;
12. }r[10000];
13. map<string,int> m;
14. int n;
15. char f[11],s[11];
16. int dp[10000][2];
17. inline int Max(int a,int b){return a>b?a:b;}
18. int dfs(int x)
19. {
20.     int i,len=r[x].son.size(),ret,sid;
21.     dp[x][1]=r[x].rp;
22.     dp[x][0]=0;
23.     for(i=0;i<len;i++)
24.         {
25.             sid=r[x].son[i];
26.             ret=dfs(sid);
27.             if(ret==1)//如果取的时候大
28.                 {
29.                     dp[x][0]+=dp[sid][1];
30.                     dp[x][1]+=Max(dp[sid][0],dp[sid][1]-r[x].sub[i]);
31.                 }
32.             else//如果不取的时候大
33.                 {
34.                     dp[x][0]+=dp[sid][0];
35.                     dp[x][1]+=dp[sid][0];
36.                 }
37.         }
38.     return (dp[x][1]>dp[x][0]?1:0);
39. }
40. int main()
41. {
42.     int i,tida,tidb,tmp,root,ans;
43.     while(scanf("%d",&n)!=EOF)
44.         {
46.             m.clear();
47.             for(i=0;i<n;i++)
48.             {
49.                 scanf("%s%d",r[i].name,&r[i].rp);
50.                 m[r[i].name]=i;
51.                 r[i].son.clear();
52.                 r[i].sub.clear();
53.             }
54.             for(i=0;i<n-1;i++)
55.             {
56.                 scanf("%s%s%d",f,s,&tmp);
57.                 tida=m[f];
58.                 tidb=m[s];
59.                 if(i==0)root=tida;
60.                 r[tida].son.push_back(tidb);
61.                 r[tida].sub.push_back(tmp);
62.             }
63.             if(n==1)
64.             {
65.                 printf("%d/n",r[0].rp);
66.                 continue;
67.             }
68.             dfs(root);
69.             for(i=ans=0;i<n;i++)ans=Max(ans,dp[i][0]),ans=Max(ans,dp[i][1]);
70.             printf("%d/n",ans);
71.         }
72.     return 0;
73. }