洛谷 P1272 重建道路

题目：重建道路

思路：
树形dp。
记忆化搜索实现。
f[x][y]：第i个点为及以下需要y个点最少划几下。

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 150
#define read(x) scanf("%d",&x)

int n,p;
vector<int> g[maxn+5];
int f[maxn+5][maxn+5];
int sum[maxn+5];

void dp(int x) {
	sum[x]=1;
	for(int i=0;i<g[x].size();i++) {
		int y=g[x][i];
		dp(y);
		sum[x]+=sum[y];
		for(int j=sum[x];j>=1;j--) {
			for(int k=1;k<j;k++) {
				f[x][j]=min(f[x][j],f[x][j-k]+f[y][k]-1);
			}
		}
	}
}

int main() {
	read(n),read(p);
	for(int i=1;i<n;i++) {
		int x,y;
		read(x),read(y);
		g[x].push_back(y);
	}
	for(int i=0;i<=n;i++) for(int j=0;j<=p;j++) f[i][j]=1e9;
	for(int i=1;i<=n;i++) f[i][1]=g[i].size();
	dp(1);
	int ans=f[1][p];
	for(int i=2;i<=n;i++) ans=min(f[i][p]+1,ans);
	printf("%d",ans);
	
	return 0;
}