关闭

hdu 1561 树形dp 入门教学

标签: hdudp1561教学
216人阅读 评论(0) 收藏 举报
分类:

第一次遇到树形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]); 
	}
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:37868次
    • 积分:1370
    • 等级:
    • 排名:千里之外
    • 原创:94篇
    • 转载:4篇
    • 译文:0篇
    • 评论:14条
    最新评论