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

题目链接

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

View Code

 1 #include <stdio.h>

 2 #include <memory.h>

 3 #define MAX(a,b) ((a)>(b)?(a):(b))

 4 #define N 205

 5 int son[N],bro[N],w[N],n,m;

 6 int c[N][N];

 7 void Insert(int u,int fa,int x)

 8 {

 9   w[u]=x;

10   if(son[fa]==-1) son[fa]=u;

11   else  bro[u]=bro[son[fa]],bro[son[fa]]=u;

12 }

13 int dp(int r,int k)

14 {

15   int i,ret;

16   if(r==-1 || !k)  return 0;

17   if(c[r][k]!=-1) return c[r][k];

18   ret=dp(bro[r],k);

19   for(i=0;i<k;i++) ret=MAX(ret,w[r]+dp(son[r],i)+dp(bro[r],k-1-i));

20   return c[r][k]=ret;

21 }

22 int main()

23 {

24   int i,fa,x;

25   while(~scanf("%d%d",&n,&m)&&n)

26   {

27     memset(son,-1,sizeof(son));

28     memset(bro,-1,sizeof(bro));

29     for(i=1;i<=n;i++)

30     {

31       scanf("%d%d",&fa,&x);

32       Insert(i,fa,x);

33     }

34     w[0]=0;

35     memset(c,-1,sizeof(c));

36     printf("%d\n",dp(0,m+1));

37   }

38   return 0;

39 }