dp专辑 A - Rebuilding Roads [ 树形dp]

第二道树形dp的题，看别人的代码，看了一下午才明白，还没入门呀~

题意：

有n个点组成一棵树，问至少要删除多少条边才能获得一棵有p个结点的子树？

思路：

设dp[i][k]为以i为根，生成节点数为k的子树，所需剪掉的边数。

dp[i][1] = total(i.son) + 1，即剪掉与所有儿子（total(i.son)）的边，还要剪掉与其父亲（+1）的边。

dp[i][k] = min(dp[i][k],dp[i][j - k] + dp[i.son][k] - 2)，即由i.son生成一个节点数为k的子树，再由i生成其他j-k个节点数的子树。

这里要还原i与i.son和其父亲的边所以-2。

参考大牛的代码~  一开始

dp[root][j] = min(dp[root][j], dp[root][j - k] + dp[tree[root][i]][k] - 2);

中的-2，始终想不明白，不是-1就可以了吗？在这里纠结了好久，后来结合

for (int i = 1; i <= n; i++)
    {
//为以i为根，生成节点数为1的子树所需剪掉的边数  每个结点都有个父结点 +1  根结点有个虚拟的父结点，方便统一处理
        dp[i][1] = tree[i].size() + 1;
        for (int j = 2; j <= p; j++)
            dp[i][j] = MAX;
    }

dp[root][p]--;

才明白了，顿感大牛们智商太高了，膜拜一会儿~后来仔细想想，这样也是为了方便处理问题~

//AC CODE：

#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
using namespace std;

const int MAX = 10000;

vector <int> tree[160];
//设dp[i][k]为以i为根，生成节点数为k的子树  所需 剪掉 的 边数
int a, b, n, p, dp[160][160];
bool son[160];

void dfs(int root)
{
    int len=tree[root].size();
    for (int i = 0; i < len; i++)
    {
        dfs(tree[root][i]);//递归调用孩子结点（后根遍历）
        for (int j = p; j > 1; j--)//j==1 的情况已经存在 >1 即可
            for (int k = 1; k < j; k++)
                dp[root][j] = min(dp[root][j], dp[root][j - k] + dp[tree[root][i]][k] - 2);
    }
}

int main()
{
    scanf("%d %d",&n,&p);
    memset(son, false, sizeof(son));
    for (int i = 0; i < n - 1; i++)
    {
        scanf("%d %d",&a,&b);
        tree[a].push_back(b);
        son[b] = true;//记录b是否有儿子
    }
    int root = 1;
    while(son[root])//找父结点
        root++;
    for (int i = 1; i <= n; i++)
    {
//为以i为根，生成节点数为1的子树所需剪掉的边数  每个结点都有个父结点 +1  根结点有个虚拟的父结点，方便统一处理
        dp[i][1] = tree[i].size() + 1;
        for (int j = 2; j <= p; j++)
            dp[i][j] = MAX;
    }
    dfs(root);
    dp[root][p]--;// 与dp方程中+2有关，还原i与其父亲的边，最后i为父节点，则-1
    int ans = MAX;
    for (int i = 1; i <= n; i++)
        ans = min(ans, dp[i][p]);
    printf("%d\n",ans);
    return 0;
}