HDOJ树形DP专题之The more, The Better

题目链接

这题要用左二子，右兄弟的存储结构来存树（附加一个结点0，将森林连成树），然后就是在二叉树上DP。

#include <stdio.h>
#include <memory.h>
#define MAX(a,b) ((a)>(b)?(a):(b))
#define N 205
int son[N],bro[N],w[N],n,m;
int c[N][N];
void Insert(int u,int fa,int x)
{
  w[u]=x;
  if(son[fa]==-1) son[fa]=u;
  else  bro[u]=bro[son[fa]],bro[son[fa]]=u;
}
int dp(int r,int k)
{
  int i,ret;
  if(r==-1 || !k)  return 0;
  if(c[r][k]!=-1) return c[r][k];
  ret=dp(bro[r],k);
  for(i=0;i<k;i++) ret=MAX(ret,w[r]+dp(son[r],i)+dp(bro[r],k-1-i));
  return c[r][k]=ret;
}
int main()
{
  int i,fa,x;
  while(~scanf("%d%d",&n,&m)&&n)
  {
    memset(son,-1,sizeof(son));
    memset(bro,-1,sizeof(bro));
    for(i=1;i<=n;i++)
    {
      scanf("%d%d",&fa,&x);
      Insert(i,fa,x);
    }
    w[0]=0;
    memset(c,-1,sizeof(c));
    printf("%d\n",dp(0,m+1));
  }
  return 0;
}