hdu1054 Strategic Game(树形DP)

嘿嘿,我的第一道树形DP

题意:用最少的点覆盖整棵树,看一下Sample Input 就知道题意。

分析:

在树上做一个动态规划

【状态】:

dp[i][0] 为以 i 为根节点,并且该节点不放,所需要的最少的点数

dp[i][1] 为以 i 为根节点,并且该节点放,所需要的最少的点数

【转移方程】:

dp[i][0]=sum(dp[son[i][j]][1]) 该点不放的话,那么它的儿子节点必须都放,这样之间的边才可以被覆盖

dp[i][1]=sum(min(dp[son[i][j]][0],dp[son[i][j]][1])) 该点放的话,那么它的儿子节点就有两种决策,一种是放,一种是不放,取 min 就行了

#include<iostream>
using namespace std;
int dp[1505][2],f[1505],ans,n;
int son[1505][1505],size[1505];
int dfs(int pos,int val)
{
	if(dp[pos][val]!=INT_MIN)
		return dp[pos][val];
	int sum=val;
	for(int i=0;i<size[pos];i++)
	{
		if(val==1)
			sum+=min(dfs(son[pos][i],0),dfs(son[pos][i],1));
		else sum+=dfs(son[pos][i],1);//当前节点不选,则子节点必须要选
	}
	return dp[pos][val]=sum;
}
int main()
{
	while(scanf("%d",&n)==1)
	{
		for(int i=0;i<n;i++)
		{
			f[i]=i;
			dp[i][1]=dp[i][0]=INT_MIN;
		}
		for(int i=0;i<n;i++)
		{
			int x,m;
			scanf("%d:(%d)",&x,&m);
			size[x]=m;
			for(int j=0;j<size[x];j++)
			{
				scanf("%d",&son[x][j]);
				f[son[x][j]]=x;
			}
		}
		for(int i=0;i<n;i++)
		{
			if(f[i]==i)
			{
				ans=min(dfs(i,0),dfs(i,1));
				break;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}
原文地址:https://www.cnblogs.com/nanke/p/2269484.html