【ACWing】1252. 搭配购买

题目地址：

https://www.acwing.com/problem/content/1254/

Joe觉得云朵很美，决定去山上的商店买一些云朵。商店里有$n$朵云，云朵被编号为$1,2,\dots ,n$，并且每朵云都有一个价值。但是商店老板跟他说，一些云朵要搭配来买才好，所以买一朵云则与这朵云有搭配的云都要买。但是Joe的钱有限，所以他希望买的价值越多越好。

输入格式：
第$1$行包含三个整数$n$，$m$，$w$，表示有$n$朵云，$m$个搭配，Joe有$w$的钱。第$2\sim n+1$行，每行两个整数$c_i$，$d_i$表示$i$朵云的价钱和价值。第$n+2\sim n+1+m$行，每行两个整数$u_i$，$v_i$，表示买$u_i$就必须买$v_i$，同理，如果买$v_i$就必须买$u_i$。

输出格式：
一行，表示可以获得的最大价值。

数据范围：
$1\le n\le 10000$
$0\le m\le 5000$
$1\le w\le 10000$
$1\le c_i\le 5000$
$1\le d_i\le 100$
$1\le u_i,v_i\le n$

由于某个组合必须一起买，所以想到将每个组合捏成一个物品，算出其价格和价值，然后用$0-1$背包问题的求解方式求解。捏成一个物品的过程可以用并查集来做，在合并的时候，同时要记录每个集合的总价格和总价值。并且在做动态规划的时候，只考虑根节点的信息（根节点存的就是其所在集合的总价格和总价值）。代码如下：

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

const int N = 10010;
int n, m, vol;
int v[N], w[N];
int p[N];
int f[N];

int find(int x) {
    if (x != p[x]) p[x] = find(p[x]);
    return p[x];
}

int main() {
    scanf("%d%d%d", &n, &m, &vol);
    for (int i = 1; i <= n; i++) p[i] = i;
    for (int i = 1; i <= n; i++) scanf("%d%d", &v[i], &w[i]);

    while (m--) {
        int a, b;
        scanf("%d%d", &a, &b);
        int pa = find(a), pb = find(b);
        if (pa != pb) {
            p[pa] = pb;
            v[pb] += v[pa];
            w[pb] += w[pa];
        }
    }
    
    for (int i = 1; i <= n; i++)
        if (p[i] == i)
        	// 0-1背包里，价格要从大到小遍历
            for (int j = vol; j >= v[i]; j--)
                f[j] = max(f[j], f[j - v[i]] + w[i]);

    printf("%d\n", f[vol]);

    return 0;
}

时间复杂度$O(n+m{\log }^{\ast }n+nw)$，空间$O(n)$。