poj 1947(树形dp)

题意：给出一颗树，求解截出k个连通点需要去掉的最少边树

思路：树形dp，dp[i][j] 表示以编号为i的点为根的子树包含j个连通点的最小代价。其中dp[i][1]就是该顶点的分支个数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define inf 0x3f3f3f
#define N 155
vector<int> g[N];
int dp[N][N];
int m,mi;
void dfs(int u){
    for(int i=0;i<=m;i++){
        dp[u][i]=inf;
    }
    dp[u][1]=g[u].size();
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i];
        dfs(v);
        for(int j=m;j>=2;j--){
            for(int k=j-1;k>=1;k--){
                dp[u][j]=min(dp[u][j],dp[u][k]+dp[v][j-k]-1);
            }
        }
    }
    if(u==1)dp[u][m]--;
    mi=min(dp[u][m]+1,mi);
}
int main(){
    int n;
    while(scanf("%d%d",&n,&m)!=EOF){
        memset(g,0,sizeof(g));
        for(int i=1;i<n;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            //g[y].push_back(x);
        }
        mi=inf;
        dfs(1);
        printf("%d\n",mi);
    }
    return 0;
}