洛谷 P1272 重建道路 (树形dp + 背包)

题意：给你一棵树，问你最少切几条边使得最后的子树有恰好有p个顶点。

思路：我们设dp[i][j]表示的是你在i顶点保留j个点的最小切割数量，那么

dp[u][j] = max(dp[u][j],dp[u][k] + dp[v][j-k]) ,初始化是 dp[u][1] = size[v];

代码：

#include <bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
const int maxn = 155;
vector<int>G[maxn];
int Size[maxn] , dp[maxn][maxn];
int ans = INF , n , p;
void dfs(int u,int fa)
{
    dp[u][1] = Size[u];
	for(auto i : G[u])
    {
        int v = i;
        if(v == fa) continue;
        dfs(v,u);
        for(int j = p ; j >= 2 ; j --)
        {
            for(int k = 1 ; k < j ; k++)
            {
                dp[u][j] = min(dp[u][j] , dp[u][k] + dp[v][j-k] - 2);
            }
        }
    }
    ans = min(ans,dp[u][p]);
}
int main()
{
    int u ,v;
	scanf("%d%d",&n,&p);
	memset(dp,INF,sizeof(dp));
	for(int i = 0 ; i < n - 1 ; i++)
	{
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
        Size[u] ++;
		Size[v] ++;
	}
	dfs(1,0);
	cout<<ans<<endl;
}