链接:http://acm.hdu.edu.cn/showproblem.php?pid=1520
题意:一堆人要聚会,为了避免不愉快,所以员工和他的直接上司不能同时出席,每个人出席的话都会有一个值,问怎么安排,是这个值最大~
思路:整个关系构成了一棵树,dp方程也很简单,看出来是树状dp的话就很简单了= =。。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<iomanip>
#define INF 99999999
using namespace std;//一堆头文件。。
const int maxn=6000+10;
int n,top,head[maxn],pre[maxn];//pre保存父节点是谁
int num[maxn];
int dp[maxn][2];
struct Side{//链式前向星存树(其实就是存图,只不过这个图是树)
int v,next;
}side[maxn];
void init(){
memset(head,-1,sizeof(head));
memset(pre,0,sizeof(pre));
memset(dp,0,sizeof(dp));
top=0;
}
void add_side(int u,int v){
side[top]=(Side){v,head[u]};
head[u]=top++;
}
void dfs(int u){//搜索 答案
for(int i=head[u];i != -1;i=side[i].next){
int v=side[i].v;
dfs(v);
dp[u][0] += max(dp[v][0],dp[v][1]);
dp[u][1] += dp[v][0];
}
dp[u][1] += num[u];
}
int main(){
int u,v;
while(~scanf("%d",&n)){
init();
for(int i=1;i<=n;++i)
scanf("%d",&num[i]);
while(~scanf("%d%d",&v,&u),u+v)
pre[v]=u;
for(int i=1;i<=n;++i)
add_side(pre[i],i);
dfs(0);//假设0节点是大boss的“上司”
printf("%d\n",dp[0][0]);
}
return 0;
}