【树形DP】 POJ 1947 Rebuilding Roads

点击打开链接

最少剪去多少条边使得存在一个节点数为P的树

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#include<queue>
#include <string>
#include <sstream>
#include <map>
#include <vector>
#define LL long long
#define N 1000100
const int INF= 1e8;
vector<int>g[191];
int dp[189][192];//1 节点 ，2 算上该节点有多少点
int n,p,x,y,num[200];
int dfs(int u,int pre)
{
    num[u]=1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==pre) continue;
        num[u]+=dfs(v,u);
        for(int j=num[u];j>1;j--)
        {
            for(int k=1;k<j;k++)
            {
                dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[v][k]-1);//表示 不剪这条边 先前已经剪过所以需要-1
            }
        }
    }
    return num[u];
}
int main()
{
    while(scanf("%d%d",&n,&p)!=EOF)
    {
        for(int i=0;i<=n;i++) g[i].clear();
        for(int i=0;i<n-1;i++)
        {
            cin>>x>>y;
            g[x].push_back(y);
        }
//        memset(dp,-1,sizeof(dp));
        for(int i=0;i<=n;i++)
            for(int j=0;j<=n;j++) dp[i][j] = INF;
        for(int i=0;i<=n;i++) dp[i][1]=g[i].size();//初始化
        dfs(1,1);
        int ans=dp[1][p];
        for(int i=2;i<=n;i++) ans=min(ans,dp[i][p]+1);
        cout<<ans<<endl;
    }
    return 0;
}