[洛谷 P1273] 有线电视网 (树形分组背包)

[洛谷 P1273] 有线电视网

---

题目链接

---

大致题意:

n个节点,m个叶子节点,根节点为1号点,每一条边都有一个边权,代表花费,叶子节点具有点权,代表收益

问在总收益>=0的情况下,可以连通最多多少个叶子节点

如果我简化题意不是很清楚,请看原题意

---

解题思路:

这个题不是很好想,如果我们把每棵子树看作是一个分组,那么组内可以不选叶子节点,可以选一个,可以选两个…

也就是说,每一个组内最多只有一种决策,分组背包!!

分组背包的状态表示是前i个组中,选j个物品的最大收益

但是我们这是树形结构,要在加一维状态,就是根节点,即f[i][u][j] 表示以u为根,在前i个子树中,选了j个物品的最大收益,其中子树代表每组物品,要选的叶子节点数是容量,当前子树选几个叶子节点是这组要选哪些物品

这样三维状态表示会MLE,所以我们要优化掉一维

我们发现,树形dp中,先对v进行dp,然后是u,也就是说在树形dp里,u的前一个状态就是v,我们就可以把第一维状态滚动优化掉

即f[u][j] 表示以u为根,选择j个的最大收益

转移方程就是分组背包,代码中带有详细注释

---

AC代码:

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const int N = 3010;

int n, m;
int a[N];
vector<pair<int, int>>e[N];
int f[N][N]; //表示以i为根,选择j个的最大收益
//f[i][u][j] 表示以u为根,在前i个子树中,选择了j个叶子节点的最大收益 滚动优化掉第一维
int dfs(int u, int fa) {
	if (u > n - m) {    //叶子节点
		f[u][1] = a[u]; //盈利
		return 1;
	}
	int sum = 0;//背包容量
	for (auto& op : e[u]) {	//每一棵子树是一个分组
		int v = op.first, w = op.second; //w是花费
		if (v == fa)continue;
		int cnt = dfs(v, u); //子树的物品个数
		sum += cnt;
		for (int i = sum; i >= 0; --i) //枚举背包容量
			for (int j = 0; j <= cnt && j <= i; ++j) //枚举物品个数 可以拿1个,2个,3个...
				f[u][i] = max(f[u][i], f[u][i - j] + f[v][j] - w);
	}
	return sum;
}
int main(void)
{
	cin >> n >> m;
	for (int i = 1; i <= n - m; ++i) {
		int k, b, c; cin >> k;
		while (k--) {
			cin >> b >> c;
			e[i].push_back({ b,c });
			e[b].push_back({ i,c });
		}
	}
	for (int i = n - m + 1; i <= n; ++i)cin >> a[i];

	memset(f, -0x3f3f3f3f, sizeof f);
	for (int i = 1; i <= n; ++i)f[i][0] = 0;

	dfs(1, -1);

	int res = 0;
	for (int i = m; i >= 0; --i)
		if (f[1][i] >= 0) {
			res = i;
			break;
		}

	cout << res << endl;
	return 0;
}