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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/imzxww/article/details/81670046

https://www.luogu.org/problemnew/show/P1272

思路:

假设把所有的边先都删除,那么我们要考虑这条边是不是要加进去,这样就变成01背包的题目了。

代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=2e2+10;
const int inf=0x3f3f3f3f;
vector<int>e[maxn];
int n,p;
int in[maxn];
int ans;
int dp[maxn][maxn];
void dfs(int x,int pre)
{
    dp[x][1]=in[x];
    for(int i=0;i<e[x].size();i++)
    {
        int v=e[x][i];
        if(v==pre)continue;
        dfs(v,x);
        for(int j=p;j>=1;j--)
        {
            for(int k=1;k<=j;k++)
            {
                dp[x][j]=min(dp[x][j],dp[x][j-k]+dp[v][k]-2);
            }
        }
    }
    ans=min(dp[x][p],ans);
}
int main()
{
    scanf("%d%d",&n,&p);
    int x,y;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        e[x].push_back(y);
        e[y].push_back(x);
        in[x]++,in[y]++;
    }
    ans=inf;
    memset(dp,inf,sizeof(dp));
    dfs(1,-1);
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/imzxww/article/details/81670046
今日推荐