hdu 1561 树形dp 入门教学

第一次遇到树形DP，研究了很久。

#include<iostream>
#include<cstdio>
#include<vector>
#include<string.h> 
using namespace std;
int m,n,value[205],dp[205][205];//dp[i][j]表示从根节点i出发，攻占j个城堡所能得到的最大价值。
vector<int>list[205];
void DFS(int root,int M)
{
	int i,j,k;
	int len = list[root].size();
	dp[root][1] = value[root];//只占领根节点的情况
	for(i = 0; i < len; i++)//从只有一个子节点的情况开始计算，一直递推到有len个子节点的情况
	{
		if(M > 1) DFS(list[root][i],M-1);
		for(j=M; j > 1; j--)//从root开始（包括攻击root），攻击M个城堡的情况
		{
			for(k = 1; k < j; k++)//k表示攻击以i为根节点的树的次数
			{
				if(dp[root][j] < dp[root][j - k] + dp[list[root][i]][k])
				dp[root][j] = dp[root][j - k] + dp[list[root][i]][k];
			}
		}
	}
}
int main()
{
	int i,j,fa;
	while(scanf("%d%d",&n,&m),m || n)
	{
		memset(dp,0,sizeof(dp));
		memset(value,0,sizeof(value));
		for(i = 0; i <= n; i++)
		{
			list[i].clear();
		}
		for(i = 1; i <= n; i++)
		{
			scanf("%d%d",&fa,&value[i]);
			list[fa].push_back(i); //将i保存为一个fa的子节点。
		}
		DFS(0,m+1);//这里要m+1而不是m，是因为我们自己额外添加了一个本来不存在的 城堡0，其价值为0，并且是整棵树的根节点。
		printf("%d\n",dp[0][m+1]); 
	}
}