Rebuilding Roads POJ - 1947

http://poj.org/problem?id=1947

dp[i][j]代表以i为根的子树 保留j个节点最少需要删几条边

详见注释

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=0x3f3f3f3f;
const int maxn=2e2+10;

int dp[maxn][maxn],e[maxn][maxn];
int book[maxn];
int n,p;

void dfs(int cur)
{
    int i,j,k;
    dp[cur][1]=0;//一开始cur之下没有子节点 所以只保留一个点不需要删边
    for(i=1;i<=n;i++){
        if(e[cur][i]){
            dfs(i);
            for(j=p;j>=1;j--){
                if(dp[cur][j]!=N){
                    dp[cur][j]++;//新增的子节点还没来得及更新答案 即dp[cur][j]还是从之前的子节点中得来 默认要切掉新节点
                }
                for(k=1;k<=j-1;k++){
                    dp[cur][j]=min(dp[cur][j],dp[cur][k]+dp[i][j-k]);//松弛答案
                }
            }
        }
    }
}

int main()
{
    int i,u,v,ans;
    scanf("%d%d",&n,&p);
    for(i=1;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        e[u][v]=1;
        book[v]=1;
    }
    memset(dp,0x3f,sizeof(dp));
    for(i=1;i<=n;i++){
        if(!book[i]){
            dfs(i);
        }
    }
    ans=N;
    for(i=1;i<=n;i++){
        if(!book[i]) ans=min(ans,dp[i][p]);
        else ans=min(ans,dp[i][p]+1);
    }
    printf("%d\n",ans);
    return 0;
}