POJ - 1947Rebuilding Roads(树形dp)

题目 给你n个节点的一颗树,问你最少切断几条边使得能独立出一个节点数为m的子树。

dp[i][j]表示生成一个以i为根有j个节点的子树需要切断的最少的边数

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=155;
int n,m,dp[maxn][maxn],siz[maxn],tmp[maxn];
vector<int>a[maxn];
void dfs(int u,int fa)
{
    dp[u][0]=1;//这个子树上一个点也没有,直接切断根与父节点一条的边
    dp[u][1]=0;//暂时是0,下面更新
    siz[u]=1;
    for(int i=0;i<a[u].size();i++)
    {
        int v=a[u][i];
        if(v==fa) continue;
        dfs(v,u);
        memset(tmp,0x3f,sizeof tmp);
        for(int j=1;j<=siz[u];j++)
        for(int k=0;k<=siz[v];k++)
        {
            tmp[j+k]=min(dp[u][j]+dp[v][k],tmp[j+k]);///转移方程
        }
        siz[u]+=siz[v];
        for(int j=1;j<=siz[u];j++)
            dp[u][j]=tmp[j];
    }
}
int main()
{
    ///cout << "Hello world!" << endl;
    ///freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&n,&m))
    {
        int x,y;
        for(int i=1;i<=n;i++)
            a[i].clear();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&x,&y);
            a[x].push_back(y);
            a[y].push_back(x);
        }
        memset(dp,-1,sizeof dp);
        dfs(1,-1);
        int ans=dp[1][m];
        for(int i=2;i<=n;i++)
        if(dp[i][m]>=0)//太智障了,一开始写成了>0...
            ans=min(ans,dp[i][m]+1);
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/dllpxfire/article/details/81063085