POJ--1947[Rebuilding Roads] 树形DP

题意:

给你一棵树,让你求最少剪掉多少条边可以剪出一棵边数为p的子树.

 

思路:(树形DP)

(1):f[i][j]:表示以i为根(包含i),共j个节点需要切断的路的数量
(2):f[i][j+k]=min(f[i][j+k],f[i.son[t]][k])(t为i的孩子)

 

 

CODE:

/*AC代码:0ms*/
#include <iostream>
#include <cstdio>
#include <memory.h>
#include <algorithm>
#include <vector>
#define MAXN 155
#define INF 1e8
#define min(a,b) (a<b?a:b)
using namespace std;
int f[MAXN][MAXN];
int N,P;
vector<int>Tree[MAXN];
void Init()
{
	int i,j,u,v;
	for(i=0;i<=N;i++)
		Tree[i].clear();
	for(i=1;i<N;i++)
	{
		scanf("%d%d",&u,&v);
		Tree[u].push_back(v);
	}
	for(i=1;i<=N;i++)
	{
		for(j=1;j<=N;j++)
			f[i][j]=INF;
	}
}
void DFS(int root)
{
	int i,j,k;
	int size=Tree[root].size();
	for(i=0;i<size;i++)
		DFS(Tree[root][i]);
	f[root][1]=size;
	//类似背包:一颗颗把支添加上去
	for(i=0;i<size;i++)
	{
		for(k=P-1;k>=1;k--)
		{
			if(f[root][k]<INF)
			{
				for(j=1;j<=P-1&&j+k<=P;j++)
				{
					if(f[Tree[root][i]][j]<INF)
						f[root][k+j]=min(f[root][k+j],f[root][k]+f[Tree[root][i]][j]-1);
				}
			}
		}
	}
}
void Solve()
{
	int i,res;
	res=INF;
	DFS(1);
	for(i=2;i<=N;i++)
	{res=min(res,f[i][P]+1);}
	res=min(res,f[1][P]);
	printf("%d\n",res);
}
int main()
{
	while(scanf("%d%d",&N,&P)!=EOF)
	{
		Init();
		Solve();
	}
return 0;
}

你可能感兴趣的:(POJ--1947[Rebuilding Roads] 树形DP)