POJ 1947 Rebuliding Roads 树形DP

题意:给出一棵树,问至少删除几条边,才能出现大小为P的子树。
思路:删边类型的树形DP。
为了保持连通性,我们定义状态dp[u][j]为以u为根的子树,节点u必须选,得到大小为j的子树所删除的最少的边。因为无法直接将删边的信息汇总到根节点上,我们最后枚举节点,求出最后的值。
下面考虑状态转移:
对于节点u的儿子节点v,我们有两种选择:
1.不删除u,v之间的边,这个时候

dp[u][j]=min(dp[v][k]+dp[u][jk]),1kj 

2.删除u,v,之间的边,则
dp[u][j]=dp[u][j]+1 

j的范围: 1jP 
边界条件:
dp[u][1]=0 

最终的答案我们要枚举节点u,同时还要注意,对于不是根结点的节点,我们还要断开该节点和父亲节点的边。
代码如下:

#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int MAX = 200;

int N,P;
int to[MAX<<1], nxt[MAX<<1],head[MAX],tot;
int dp[MAX][MAX];

void init()
{
    memset(head,-1,sizeof(head));
    memset(dp,0x3f,sizeof(dp));
    tot = 0;
}

void addedge(int u, int v)
{
    to[tot] = v , nxt[tot] = head[u];
    head[u] = tot++;
}

void dfs(int u, int p)
{
    dp[u][1] = 0;
    for(int i = head[u]; ~i; i = nxt[i]){
        int v = to[i];
        if(v == p) continue;
        dfs(v,u);
        for(int j = P; j >= 1; --j){
            dp[u][j] += 1;
            for(int k = 0; k < j; --k)
                dp[u][j] = min(dp[u][j],dp[v][k] + dp[u][j-k]);
        }
    }
}

int main(void)
{
    //freopen("input.txt","r",stdin);
    init();
    scanf("%d%d",&N,&P);
    for(int i = 0; i < N - 1; ++i){
        int u, v;
        scanf("%d%d",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    dfs(1,0);
    int ans = dp[1][P];
    for(int i = 2; i <= N; ++i)
        ans = min(ans,dp[i][P] + 1);
    printf("%d\n",ans);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值