并查集+01背包(滚动数组)
PS:
刚开始并没有想到这是一个并查集+01背包,我就以为是一个简单的并查集然后贪心就完事了,但是当时纠结的是,两边都可以贪心(精力和利益值),当时没多想随便找一个贪心然后代码过了样例,交上去的时候却wa了,这时候就开始认真分析问题了。
求解最大值有两种方式,一个是二分,一个是动态规划,这里毫无疑问是动态规划,所有的条件里面选一个最大的,这里我们用滚动数组极大的简化了代码,也使程序看起来通俗易懂,具体注释请看代码
#include <bits/stdc++.h> using namespace std; const int maxn = 1e6 + 10; int pre[maxn], a[maxn], b[maxn], dp[maxn]; int find(int x) { if (x != pre[x]) pre[x] = find(pre[x]); return pre[x]; } int main() { int t, n, m, c, ans, cnt; scanf("%d", &t); while (t--) { ans = 0, cnt = 0; scanf("%d%d%d", &n, &m, &c); for (int i = 1; i <= n; ++i) pre[i] = i; for (int i = 2; i <= n; ++i) scanf("%d%d", &a[i], &b[i]); for (int j = 1; j <= m; ++j) { int x, y; scanf("%d%d", &x, &y); pre[find(x)] = find(y); //合并操作 } memset(dp, 0, sizeof(dp)); for (int i = 2; i <= n; ++i) if (find(i) == find(1)) //在满足条件的情况下才去dp { for (int j = c; j >= a[i]; --j) dp[j] = max(dp[j], dp[j - a[i]] + b[i]); } printf("%d\n", dp[c]); } }