洛谷 ~ P1272 ~ 重建道路(树形背包)

在这里插入图片描述

思路

我们转化一下,假设把所有的边先都删除,那么我们要考虑的就是这条边是不是要加进去,这就是一个01背包+树形dp。
定义deg表示 u u u点的度数。
定义 d p [ u ] [ j ] dp[u][j] dp[u][j]表示以 u u u为根保留 j j j个结点所需要删除的边数。
d p [ u ] [ 1 ] = d e g [ u ] dp[u][1]= deg[u] dp[u][1]=deg[u]
d p [ u ] [ j ] = m i n ( d p [ u ] [ j ] , d p [ u ] [ i ] + d p [ v ] [ j − i ] − 2 ) , ( j : p − > 2 ) ( i : 1 − > j − 1 ) , v 是 u 的 儿 子 dp[u][j] = min(dp[u][j],dp[u][i]+dp[v][j-i]-2),(j: p->2) \quad (i:1->j-1),v是u的儿子 dp[u][j]=min(dp[u][j],dp[u][i]+dp[v][ji]2)(j:p>2)(i:1>j1)vu
为什么要 -2 呢?因为 d p [ u ] [ i ] dp[u][i] dp[u][i]中把u->v这条边删掉了, d p [ v ] [ j − i ] dp[v][j-i] dp[v][ji]中也把u->v这条边删掉了。
a n s = m i n ( d p [ 1 − > n ] [ p ] ) ans=min(dp[1->n][p]) ans=min(dp[1>n][p])
记得初始化dp为INF

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 155;
const int INF = 0x3f3f3f3f;
int n, p, deg[MAXN], dp[MAXN][MAXN];
vector<int> G[MAXN];
void dfs(int u, int fa)
{
    dp[u][1] = deg[u];
    for (auto v : G[u])
    {
        if (v == fa) continue;
        dfs(v, u);
        for (int j = p; j >= 2; j--)
            for (int i = 1; i < j; i++)
             dp[u][j] = min(dp[u][j], dp[u][i] + dp[v][j-i] - 2);
    }
}
int main()
{
    scanf("%d%d", &n, &p);
    memset(dp, INF, sizeof(dp));
    for (int i = 0; i < n-1; i++)
    {
        int u, v; scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
        deg[u]++, deg[v]++;
    }
    dfs(1, -1);
    int ans = INF;
    for (int i = 1; i <= n; i++) ans = min(ans, dp[i][p]);
    printf("%d\n", ans);
    return 0;
}
/*
11 6
1 2
1 3
1 4
1 5
2 6
2 7
2 8
4 9
4 10
4 11
*/

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值