POJ题目1947 Rebuilding Roads（树形dp）

Description

Input

Output

Sample Input

11 6
1 2
1 3
1 4
1 5
2 6
2 7
2 8
4 9
4 10
4 11

Sample Output

2

Hint

Source

USACO 2002 February

题目大意：问一个数删掉最少条边变成一个仅仅有n个结点的子树

ac代码

#include<stdio.h>
#include<string.h>
#define min(a,b) (a>b?
b:a)
#define INF 0xfffffff
int dp[220][220];
int pre[220],head[220],vis[220],dig[220];
int n,p,cnt;
struct s
{
	int u,v,w,next;
}edge[220*2];
void add(int u,int v)
{
	edge[cnt].u=u;
	edge[cnt].v=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void tree_dp(int u)
{
	int i,j,k;
	for(i=0;i<=p;i++)
	{
		dp[u][i]=INF;
	}
	dp[u][1]=0;
	for(i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;
		tree_dp(v);
		for(k=p;k>=1;k--)
		{
			dp[u][k]=dp[u][k]+1;
			for(j=1;j<k;j++)
			{

				dp[u][k]=min(dp[u][k],dp[u][j]+dp[v][k-j]);
			}
		}
	}
}
int DP(int u)
{
	tree_dp(u);
	int ans=dp[u][p];
	int i;
	for(i=1;i<=n;i++)
	{
		ans=min(ans,dp[i][p]+1);
	//	printf("%d\n",dp[i][1]);
	}
	return ans;
}
int main()
{
	//int n,p;
	while(scanf("%d%d",&n,&p)!=EOF)
	{
		int i;
		memset(dig,0,sizeof(dig));
		memset(head,-1,sizeof(head));
		cnt=0;
		for(i=0;i<n-1;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			add(a,b);
			dig[b]++;
		}
		int root;
		for(i=1;i<=n;i++)
		{
			if(dig[i]==0)
				root=i;
		}
		printf("%d\n",DP(root));
	}
}