P1272 重建道路

P1272 重建道路

大意

求一颗树上，分离出来 \(p\) 个点，使得其所需减少的边最小，求这个值。

思路

考虑树上背包。

定义状态 \(f_{u, j}\) 表示 \(u\) 节点，切掉 \(j\) 个所需的最小花费。

那么我们的初始状态就是 \(f_{u, 0} = 0\), \(f_{u, sz_u} = 1\)。

然后我们考虑如何进行转移：

\[f_{u, j} = \min(f_{u, j}, f_{u, j - k} + f_{v, k}) \]

然后考虑我们的答案如何进行计算，显然是 \(f_{u, sz_u - p} + f_{u, sz_u}\)，这个地方的 \(f_{u, sz_u}\) 的值显然是 \(1\)，但是只有根节点是 \(0\)。

这个地方实际上就是为了让你的选出来的这 \(p\) 个点与你原来的 \(u\) 子树脱离，显然需要你删一条父边。

代码

#include<iostream>
#include<cstring>
using namespace std;

const int MAXN = 155;

struct node{
	int to, nxt;
}e[MAXN];
int sz[MAXN], n, p;
int h[MAXN], tot = 0;
void add(int x, int y){
	e[++ tot] = {y, h[x]};
	h[x] = tot;
}

int f[MAXN][MAXN];

void dfsz(int u, int fa){
	sz[u] = 1;
	for(int i = h[u];i;i = e[i].nxt){
		int v = e[i].to;
		if(v == fa) continue;
		dfsz(v, u);
		sz[u] += sz[v];
	}
	f[u][0] = 0;
	f[u][sz[u]] = 1;
}

int ans = 1e9;

void dfs(int u, int fa){
	for(int i = h[u];i;i = e[i].nxt){
		int v = e[i].to;
		if(v == fa) continue;
		dfs(v, u);
		for(int j = sz[u] - 1;j >= 0;j --){
			for(int k = 0;k <= j;k ++){
				f[u][j] = min(f[u][j], f[u][j - k] + f[v][k]);
			}
		}
	}
	if(sz[u] - p >= 0){
        if(u == 1) ans = min(ans, f[u][sz[u] - p]);
        else ans = min(ans, f[u][sz[u] - p] + 1);
	}
}

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n >> p;
	
	for(int i = 1;i < n;i ++){
		int u, v; cin >> u >> v;
		add(u, v); add(v, u);
	}
	memset(f, 0x3f3f3f, sizeof(f));
	dfsz(1, 0);
	dfs(1, 0);
	cout << ans << '\n';
	return 0;
}