重建道路 洛谷 - P1272

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

dp[cur][i]代表对以cur为根的子树删去i个点的最小代价 当cur的所有子节点v的信息（即dp[v][i]）已知后 就是一个背包问题 容量就是cur子树的大小 每件物品的重量就是i 价值就是dp[v][i] 注意这里是分组背包 即每棵子树一个组 对每棵子树只能选一件物品

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

struct node
{
    int v,next;
};

node edge[maxn];
int dp[maxn][maxn];
int first[maxn],book[maxn],sum[maxn];
int n,p,num,root,ans;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void dfsI(int cur)
{
    int i,v;
    sum[cur]=1;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        dfsI(v);
        sum[cur]+=sum[v];
    }
}

void dfsII(int cur)
{
    int tmp[maxn];
    int i,j,k,v;
    dp[cur][0]=0,dp[cur][sum[cur]]=1;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        dfsII(v);
        for(k=0;k<=sum[cur];k++) tmp[k]=dp[cur][k];
        for(j=0;j<=sum[v];j++){
            for(k=sum[cur];k>=j;k--){
                tmp[k]=min(tmp[k],dp[cur][k-j]+dp[v][j]);
            }
        }
        for(k=0;k<=sum[cur];k++) dp[cur][k]=tmp[k];
    }
    if(sum[cur]>=p){
        if(cur==root) ans=min(ans,dp[cur][sum[cur]-p]);
        else ans=min(ans,dp[cur][sum[cur]-p]+1);
    }
}

int main()
{
    int i,u,v;
    scanf("%d%d",&n,&p);
    memset(first,-1,sizeof(first));
    num=0;
    for(i=1;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        addedge(u,v);
        book[v]=1;
    }
    for(i=1;i<=n;i++){
        if(!book[i]) root=i;
    }
    dfsI(root);
    memset(dp,0x3f,sizeof(dp));
    ans=n;
    dfsII(root);
    printf("%d\n",ans);
    return 0;
}

/*
5 2
1 2
1 3
1 4
1 5
*/