最勇敢的机器人（并查集+分组背包）

题目描述

机器人们都想知道谁是最勇敢的，于是它们比赛搬运一些物品。
它们到了一个仓库，里面有n个物品，每个物品都有一个价值Pi和重量Wi，但是有些物品放在一起会爆炸，并且爆炸具有传递性。（a和b会爆炸、b和c会爆炸则a和c会爆炸）机器人们可不想因此损失自己好不容易从Wind那里敲诈来的装备，于是它们想知道在能力范围内，它们最多可以拿多少价值的物品。
你能帮助它们吗？

输入

每组测试数据
第1行为n,Wmax,k(0<=n,Wmax,k<=1000)
接下来n行，为每个物品的Pi,Wi(0<=Pi<=1000,1<=Wi<=10,均为整数)
再接下来k行，每行2个数字a,b表示a和b会发生爆炸 

输出

对每组数据输出1行，为最大可能价值 

样例输入

3 10 1
100 1
200 5
10 5
1 2

样例输出

210

思路

并查集+分组背包，基本都是套模板，这里的背包用的一维，注意j要倒序遍历

很巧妙的点，数组的第一个下标0可以存该数组的长度

另，准备改改自己的码风，还在练手速中。。。

AC代码

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
const ll mo = 998244353;
const int inf = 0x3f3f3f3f;
int n,m,k,x,y,f[1010],h[1010][1010],dp[1010];
struct data {
    int v, w;
}a[1010];
int find(int x)
{
    if(x == f[x]) return x;
    return f[x] = find(f[x]);
}
int main()
{
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin >> n >> m >> k;
    for (int i = 1; i <= n; i++) {
        cin >> a[i].v >> a[i].w;
    }
    for (int i = 1; i <= n; i++) f[i] = i;
    while (k--){
        cin >> x >> y;
        f[find(x)] = find(y);
    }
    for (int i = 1; i <= n; i++) {
        h[find(i)][++h[find(i)][0]] = i;
    }
    for (int i = 1; i <= n; i++) {
        if (h[i][0] == 0) continue;
        for (int j = m; j >= 0; j--) {
            for (int k = 1; k <= h[i][0]; k++) {
                int p = h[i][k];
                if(j >= a[p].w) {
                    dp[j] = max(dp[j], dp[j-a[p].w]+a[p].v);
                }
 
            }
        }
    }
    cout << dp[m];
    return 0;
}

二维分组背包的思考

一开始想写二维背包的，后来发现要考虑很多，比如要跟本组其他物品比较，还可能该组一个都不选。。但改到最后也只有八十几分，改不动了。。。

以后碰到背包还是都用一维写吧，省事~

如果有幸被大佬翻到了这篇博客，欢迎在评论区提点提点俺！

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
const ll mo = 998244353;
const int inf = 0x3f3f3f3f;
int n,m,k,x,y,f[1010],h[1010][1010],dp[1010][1010];
struct data {
    int v, w;
}a[1010];
int find(int x)
{
    if(x == f[x]) return x;
    return f[x] = find(f[x]);
}
int main()
{
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin >> n >> m >> k;
    for (int i = 1; i <= n; i++) {
        cin >> a[i].v >> a[i].w;
    }
    for (int i = 1; i <= n; i++) f[i] = i;
    while (k--){
        cin >> x >> y;
        f[find(x)] = find(y);
    }
    for (int i = 1; i <= n; i++) {
        h[find(i)][++h[find(i)][0]] = i;
    }
    for (int i = 1; i <= n; i++) {
        if (h[i][0] == 0) continue;
        for (int j = 0; j <= m; j++) {
            for (int k = 1; k <= h[i][0]; k++) {
                int p = h[i][k];
                if(j >= a[p].w) {
                    int tmp = max(dp[i-1][j], dp[i-1][j-a[p].w]+a[p].v);
                    dp[i][j] = max(dp[i][j], tmp); //与本组其他值比较
                }
            }
            dp[i][j] = max(dp[i][j], dp[i-1][j]); //该组一个都不选可能更优
        }
    }
    cout << dp[n][m];
    return 0;
}