POJ 1947 Rebuilding Roads 树形DP

题意：给出一棵树，至少切几下，能够得到一个包含p个节点的子树。
题解：假设节点u包含了cnt[u]个子节点：1,2,3，····cnt[u]，每个子节点类似于一个背包，可以加或者不加。 当所有子节点都不加的时候，仅仅包含根节点一个，即dp[u][1]。那么我们需要对dp[u][1]进行一下初始化。即所有节点都不加的时候，只要切0次就得到根节点，即dp[u][1]=0。

#include <iostream>
using namespace std;

#define N 155
#define INF 10000
#define min(a,b) (a<b?a:b)

int dp[N][N], next[N][N];/* dp[i][j]表示以i为根节点，节点总数为j的子树，需要切的次数。 next[i][j]表示节点i的第j个子节点 */
int cnt[N], pre[N]; /* cnt[i]记录节点i的子节点数量。 pre[i]记录i的双亲节点 */
int n, p;

void dfs ( int u )
{
	dp[u][1] = 0;
	int i, j, k, v, temp;
	for ( i = 1; i <= cnt[u]; i++ )
	{
		v = next[u][i];
		dfs ( v );
		for ( j = p; j >= 0; j-- ) /* 类似于背包问题，不能 0 -> p */
		{
			temp = dp[u][j] + 1;
			for ( k = 0; k <= j; k++ )
				temp = min (dp[u][k]+dp[v][j-k], temp);
			dp[u][j] = temp;
		}
	}
}

		
int main()
{
	int i, j, x, y, ans;
	while ( scanf("%d %d", &n, &p) != EOF )
	{
		memset(next,0,sizeof(next));
		memset(cnt,0,sizeof(cnt));
		memset(pre,0,sizeof(pre));
		
		for ( i = 1; i < n; i++ )
		{
			scanf("%d %d",&x,&y);
			next[x][++cnt[x]] = y;
			pre[y] = 1;
		}
		
		for ( i = 0; i <= n; i++ )
			for ( j = 0; j <= n; j++ )
				dp[i][j] = INF;
			
		for ( i = 1; i <= n; i++ )
			if ( ! pre[i] ) break;/* 双亲节点不存在的便是根节点 */
		dfs ( i );
				
/*切完后的每一个节点都没有断绝与根节点的关系，那么需要在i与他的双亲节点之间再切一次（根节点除外）*/
		ans = dp[i][p];
		for ( i = 2; i <= n; i++ )
		    if ( dp[i][p] < ans ) 
			    ans = dp[i][p] + 1;
		printf("%d\n",ans);
	}
	return 0;
}

贴一下错误的代码：

#include <iostream>
using namespace std;

#define N 155
#define INF 10000
#define min(a,b) (a<b?a:b)

int dp[N][N], next[N][N];
int cnt[N], pre[N];
int n, p;

void dfs ( int u )
{
	if ( cnt[u] == 0 ) /* WR在这里·· */
	{
		dp[u][1] = 0;
		return;
	}
	int i, j, k, v, temp;
	for ( i = 1; i <= cnt[u]; i++ )
	{
		v = next[u][i];
		dfs ( v );
		for ( j = p; j >= 0; j-- )
		{
			temp = dp[u][j] + 1;
			for ( k = 0; k <= j; k++ )
				temp = min (dp[u][k]+dp[v][j-k], temp);
			dp[u][j] = temp;
		}
	}
}

		
int main()
{
	int i, j, x, y, ans;
	while ( scanf("%d %d", &n, &p) != EOF )
	{
		memset(next,0,sizeof(next));
		memset(cnt,0,sizeof(cnt));
		memset(pre,0,sizeof(pre));
		
		for ( i = 1; i < n; i++ )
		{
			scanf("%d %d",&x,&y);
			next[x][++cnt[x]] = y;
			pre[y] = 1;
		}
		
		for ( i = 0; i <= n; i++ )
			for ( j = 0; j <= n; j++ )
				dp[i][j] = INF;
			
		for ( i = 1; i <= n; i++ )
			if ( ! pre[i] ) break;
		dfs ( i );

		ans = dp[i][p];
		for ( i = 2; i <= n; i++ )
		    if ( dp[i][p] < ans )
			    ans = dp[i][p] + 1;
		printf("%d\n",ans);
	}
	return 0;
}