POJ 1947

最新推荐文章于 2020-12-02 23:22:11 发布

YDYKL

最新推荐文章于 2020-12-02 23:22:11 发布

阅读量942

点赞数

分类专栏：树状DP 文章标签： ini

本文链接：https://blog.csdn.net/YDYKL/article/details/6674795

版权

树状DP 专栏收录该内容

0 篇文章 0 订阅

订阅专栏

http://poj.org/problem?id=1947

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<iostream>
using namespace std;
#define M 160
#define INF 1<<30

vector<int>G[M];	
int n,p;
int dp[M][M];//dp[i][j]表示以i为根节点保留j个节点需删除的最小边数 
int f[M][M];//f[i][j]表示前i个子节点 

void Update(int root)
{
	int i,j,k;
	int s=G[root].size();
	dp[root][1]=s; //如果保留的节点为1，那只需剩下root节点，需s次把root的儿子全部删去
	
	if(s==0) 
		return;//找到叶子节点了 
	
	for(i=0;i<s;i++)
		for(j=1;j<=p;j++)
			f[i][j]=INF;
	f[0][1]=1;//对于第0号儿子节点，要保留1个节点，只需把根节点与0号节点之间的边删除，即保留根节点
	
	for(i=2;i<=p;i++)
		f[0][i]=dp[G[root][0]][i-1]; //因为有root节点，所以是i-1，即f[0][i]等于root的0号节点保留i-1个节点最少所需删除的边数 
	for(i=1;i<s;i++)
	{
		for(j=1;j<=p;j++)
		{
			for(k=1;k<=j-1;k++)//用到第i号节点。前i-1号儿子节点保留k个节点，第i号儿子保留j-k个节点 
			{
				if(f[i-1][k]!=INF&&dp[G[root][i]][j-k]!=INF)
				{
					if(f[i][j]>(f[i-1][k]+dp[G[root][i]][j-k]))
						f[i][j]=f[i-1][k]+dp[G[root][i]][j-k];
				}
			}
			if(f[i-1][j]!=INF)//不用i号儿子节点。花1次代价将之删除 
			{
				if(f[i][j]>f[i-1][j]+1)
					f[i][j]=f[i-1][j]+1;
			} 
		} 
	}
	for(i=1;i<=p;i++)
	{
		dp[root][i]=f[s-1][i];//把f数组的值转移给dp 
		//printf("***%d %d\n",i,f[s-1][i]);
	}
}

void dfs(int root)
{
	int i;
	//printf("root %d   size %d\n",root,G[root].size());
	for(i=0;i<G[root].size();i++)
		dfs(G[root][i]);
	Update(root);
	return;
}

int main()
{
	int i,j;
	while(scanf("%d%d",&n,&p)!=EOF)
	{
		for(i=1;i<=n;i++)
			G[i].clear();
			
		for(i=1;i<n;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			G[a].push_back(b);
		}
		
		
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
				dp[i][j]=INF;
		dfs(1);
		
		int ans=dp[1][p];	
		for(i=2;i<=n;i++)
			if(dp[i][p]+1<ans)
			{	//printf("i=%d dp=%d\n",i,dp[i][p]);
				ans=dp[i][p]+1;
			}
		printf("%d\n",ans); 		
	
	}
	return 0;
}