[POJ 1947] Rebuilding Roads 树形dp

http://poj.org/problem?id=1947

题意:输入 n, m 表示一棵树有 n 个顶点,然后输入 n-1 条边,要求删除一些边使得树中剩下 m 个点,问最少要删除多少边。

思路:树形dp,转移方程是:
1.不加子树son,dp[rt][i] = dp[rt][i] + 1 (不加子树son,相当于删去一条边)。
2.加子树son,dp[rt][i] = min(dp[rt][j] + dp[son][i - j]) (以s为根的树的结点数加上其子树的结点数等于i)
最后答案就是在dp[i][m]中取小,要注意的一点是,如果i不是根,值还需要+1,因为要脱离原来的根,还要去掉一条边。

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>

using namespace std;

int n, m;
int dp[155][155];
vector<int>tr[155];

int Dfs(int rt, int pre)
{
    int sum = 1;
    dp[rt][1] = 0;
    int len = tr[rt].size();
    for(int i = 0; i < len; i++){
        int son = tr[rt][i];
        if(son == pre) continue;
        sum += Dfs(son, rt);
        for(int k = sum; k >= 1; k--)
        {
            dp[rt][k] = dp[rt][k] + 1; //截断子树son
            for(int l = 1; l < k; l++) 
            {
                if(dp[rt][k] > dp[rt][k-l] + dp[son][l]){
                    dp[rt][k] = dp[rt][k-l] + dp[son][l]; //加入son
                }
            }
        }
    }
    return sum;
}

int main()
{

    while(cin>>n>>m){
        int x, y;
        memset(dp, 127, sizeof(dp)); //初始化为0x7f7f7f7f
        for(int i = 0; i <= n; i++){
            dp[i][0] = 1;
            tr[i].clear();
        }
        for(int i = 1; i < n; i++){
            cin>>x>>y;
            tr[x].push_back(y);
            tr[y].push_back(x);
        }
        Dfs(1, -1);
        int minn = dp[1][m];
        for(int i = 2; i <= n; i++) //不要父结点
        {
            if(dp[i][m] + 1 < minn){
                minn = dp[i][m] + 1;
            }
        }
        cout<<minn<<endl;
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

achonor

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值