hdu1520 （树形dp）
hdu1520 http://acm.hdu.edu.cn/showproblem.php?pid=1520
题意是给定一棵树，每个结点有一个价值，要我们选择任意个结点使得总价值最大，规则是如果父亲结点被选了，那么儿子结点不可以被选，但是儿子的儿子可以被选
本来学搜索的时候找到这题搜索题，然后用搜索做的
1 #include <stdio.h> 2 #include <string.h> 3 #include <stdlib.h> 4 #include <algorithm> 5 #include <iostream> 6 #include <queue> 7 #include <stack> 8 #include <vector> 9 #include <map> 10 #include <set> 11 #include <string> 12 #include <math.h> 13 using namespace std; 14 typedef long long LL; 15 const int INF = 1<<30; 16 int val[6666]; 17 int f[6666]; 18 vector<int> g[6666]; 19 int dfs(int u, bool flag)//flag标记父亲结点是不是取 20 { 21 int t,t2; 22 if(!flag) 23 t = val[u]; 24 t2 = 0; 25 for(int i=0; i<g[u].size(); ++i) 26 { 27 int v = g[u][i]; 28 if(!flag) 29 { 30 t += dfs(v,true); 31 } 32 t2 += dfs(v,false); 33 } 34 if(flag) 35 return t2; 36 return max(t,t2); 37 } 38 int main() 39 { 40 int n,i,x,y; 41 while(scanf("%d",&n)!=EOF) 42 { 43 44 for(i=1; i<=n; ++i) 45 { 46 g[i].clear(); 47 scanf("%d",&val[i]); 48 f[i] = -1; 49 } 50 while(scanf("%d%d",&x,&y),x) 51 { 52 g[y].push_back(x); 53 f[x] = y; 54 } 55 x = 1; 56 while(f[x]!=-1) 57 x = f[x]; 58 printf("%d\n",dfs(x,false)); 59 } 60 return 0; 61 }
但是学树形dp的时候又找到这道题，然后用树形dp重新做了一次，提高上去，运行的时间变少了
dp[u][1] 表示选择结点u可以获得的最大价值
dp[u][0]表示不选择结点u可以获得的最大价值
状态转移方程是 dp[u][1] = max( dp[u][1], dp[u][1]+dp[v][0] ); dp[u][0] = max( dp[u][0],max(dp[u][0]+dp[v][1], dp[u][0]+dp[v][0]) );
1 #include <stdio.h> 2 #include <string.h> 3 #include <stdlib.h> 4 #include <algorithm> 5 #include <iostream> 6 #include <queue> 7 #include <stack> 8 #include <vector> 9 #include <map> 10 #include <set> 11 #include <string> 12 #include <math.h> 13 using namespace std; 14 typedef long long LL; 15 const int INF = 1<<30; 16 const int N = 6000 + 10; 17 int val[N]; 18 int dp[N][2];//dp[u][1]表示选这个结点的最大值， dp[u][0] 表示不选这个结点的最大值 19 struct Edge 20 { 21 int v,next; 22 }g[N<<1]; 23 int head[N],e; 24 25 void dfs(int u, int fa) 26 { 27 dp[u][1] = val[u]; 28 dp[u][0] = 0; 29 for(int i=head[u]; i!=-1; i=g[i].next) 30 { 31 int v = g[i].v; 32 if(v==fa) continue; 33 dfs(v,u); 34 dp[u][1] = max(dp[u][1],dp[u][1]+dp[v][0]); 35 dp[u][0] = max(dp[u][0],max(dp[u][0]+dp[v][1],dp[u][0]+dp[v][0] )); 36 37 //dp[u][1] += dp[v][0]; 38 //dp[u][0] += dp[v][1]; 39 } 40 } 41 void init(int n) 42 { 43 for(int i=1; i<=n; ++i) 44 head[i] = -1; 45 e = 0; 46 } 47 void addEdge(int u, int v) 48 { 49 g[e].v = v; 50 g[e].next = head[u]; 51 head[u] = e++; 52 } 53 54 int main() 55 { 56 int n,i,a,b; 57 while(scanf("%d",&n)!=EOF) 58 { 59 init(n); 60 for(i=1; i<=n; ++i) 61 scanf("%d",&val[i]); 62 while(scanf("%d%d",&a,&b),a) 63 { 64 addEdge(a,b); 65 addEdge(b,a); 66 //g[a].push_back(b); 67 //g[b].push_back(a); 68 } 69 dfs(1,-1); 70 printf("%d\n",max(dp[1][0],dp[1][1])); 71 } 72 return 0; 73 }
最近老发现用vector建图提交老师runtime error，
