QZQ-钱哲琦

每天走在疯狂追梦的大街上。我们衣衫褴褛,却毫无倦意。

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]); 
	}
}


阅读更多
版权声明:转载注明出处。有任何疑问可留言。 https://blog.csdn.net/Techmonster/article/details/49949693
文章标签: hdu dp 1561 教学
个人分类: ACM算法竞赛
上一篇hdu 1102 最小生成树 prim
下一篇hdu 1856 并查序 特殊优化
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭