树形dp入门

poj2057

某公司的上下级关系是一颗树状结构,每个人不能与他的上司同时出现,每个人有一个值,求最大值。

这个题需要注意的是如果不保存状态会超时,这似乎也是大部分dp应该注意的事情啊

#include<iostream>
#include<string.h>
#include<stdio.h>
#include<vector>
using namespace std;
const int maxa = ;
int n;
vector< int> xiashu[maxa];
int dp[maxa][];
int v[maxa];
int du[maxa];
int dfs(int x, int zai){
int maxn = ;
int sum = ;
if(dp[x][zai])
return dp[x][zai];
for(int i = ;i < xiashu[x].size(); i++){
int k = xiashu[x][i];
sum += dfs(k, );
}
maxn = sum;
if(zai)
return dp[x][zai] = maxn;
sum = ;
for(int i = ; i < xiashu[x].size(); i++){
int k = xiashu[x][i];
sum += dfs(k, );
}
return dp[x][zai] = max(maxn, sum +v[x]);
}
int main(){
while(scanf("%d", &n)!=EOF){
for(int i = ;i <= n; i++){
scanf("%d", &v[i]);
}
memset(dp, , sizeof(dp));
memset(du, , sizeof(du));
for(int i = ; i<= n; i++)
xiashu[i].clear();
int x, y;
while(){
scanf("%d%d", &x, &y);
if(!y && !x)
break;
xiashu[y].push_back(x);
du[x]++;
}
for(int i = ; i <= n; i++){
if(du[i]==)
printf("%d\n", dfs(i, ));
}
}
}
上一篇:Excel,2010,可以独立打开窗口


下一篇:解開32位元Win 7記憶體4GB限制