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;}