CODEVS 1380 没有上司的舞会

要题目吗（戳这里）

分析：一道dp模板题

先预处理每对子节点与父节点，并记录他们的关系，为接下来的dp做铺垫

然后设f[x][0]表示不取x节点，f[x][1]表示去x节点，h[x]表示x的开心指数

so we can get a conclusion:

f[x][0]=Σf[y][0](y是x的子节点) + h[x];

f[x][1]=Σmax{f[y][0],f[y][1]}

然后我们不就可以dp了吗？

P.S:我懒，就写了个记忆化搜索，大家凑合着看吧

#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;

const int N=6005;
vector<int> S[N];
int h[N],f[N][2];
bool ok[N],have[N][2];
int x,y,n;

int dfs0(int);
int dfs1(int);

int max(int a,int b){return a > b ? a : b;}

int dfs1(int x)
{
  if(have[x][1]) return f[x][1];
  int ans=h[x];
  for(int i=0;i<S[x].size();i++)
  	ans+=dfs0(S[x][i]);
  f[x][1]=ans;
  have[x][1]=true;
  return ans;
}

int dfs0(int x)
{
  if(have[x][0]) return f[x][0];
  int ans=0;
  for(int i=0;i<S[x].size();i++)
    ans+=max( dfs1(S[x][i]),dfs0(S[x][i]) );
  f[x][0]=ans;
  have[x][0]=true;
  return ans;
}

int main()
{
  scanf("%d",&n);
  for(int i=1;i<=n;i++) scanf("%d",&h[i]);
  memset(ok,true,sizeof(ok));
  memset(have,false,sizeof(have));

  while(scanf("%d%d",&x,&y) && x && y)
  {
  	S[y].push_back(x);
  	ok[x]=false;
  }
  
  int root;
  for(int i=1;i<=n;i++) if(ok[i]) {root=i;break;}
  printf("%d",max( dfs1(root) , dfs0(root) ) );
  return 0;
}