poj 2342 Anniversary party_经典树状dp

时间:2021-09-03 07:06:41

题意:Ural大学有n个职员,1~N编号,他们有从属关系,就是说他们关系就像一棵树,父节点就是子节点的直接上司,每个职员有一个快乐指数,现在要开会,职员和职员的直接上司不能同时开会,问怎才能使开会的快乐指数最高。

思路:用树状dp,dp[i][0]为不参加会议,dp[i][1]为参加会议

#include<iostream>
#include<cstdio>
#include<vector>
#define N 6010
using namespace std;
vector<int> v[N];
int dp[N][2],hp[N],mark[N];
int max(int a,int b){return a>b?a:b;}
void dfs(int k)
{
int i,len=v[k].size();
dp[k][0]=0;
dp[k][1]=hp[k];
if(len==0)
return ;
for(i=0;i<len;i++)
{
dfs(v[k][i]);
dp[k][0]+=max(dp[v[k][i]][1],dp[v[k][i]][0]);//不参加会议,下属有可以参加会议,也可以不参加会议
dp[k][1]+=dp[v[k][i]][0];//参加会议,下属不能参加会议 }
}
int main()
{
int n,a,b,i,k;
while(scanf("%d",&n)!=EOF)
{
for(i=1;i<=n;i++)
{
scanf("%d",&k);
hp[i]=k;
v[i].clear();
}
memset(mark,0,sizeof(mark));
while(scanf("%d%d",&b,&a)&&(a||b))
{
v[a].push_back(b);
mark[b]=1;
}
for(i=1;i<=n;i++)//找出根节点
if(!mark[i])
break;
dfs(i);
printf("%d\n",max(dp[i][0],dp[i][1]));
}
return 0;
}