【洛谷】P1455 搭配购买

题目地址：

https://www.luogu.com.cn/problem/P1455

题目描述：
明天就是母亲节了，电脑组的小朋友们在忙碌的课业之余挖空心思想着该送什么礼物来表达自己的心意呢？听说在某个网站上有卖云朵的，小朋友们决定一同前往去看看这种神奇的商品，这个店里有$n$朵云，云朵已经被老板编号为$1,2,3,...,n$，并且每朵云都有一个价值，但是商店的老板是个很奇怪的人，他会告诉你一些云朵要搭配起来买才卖，也就是说买一朵云则与这朵云有搭配的云都要买，电脑组的你觉得这礼物实在是太新奇了，但是你的钱是有限的，所以你肯定是想用现有的钱买到尽量多价值的云。

输入格式：
第一行输入三个整数，$n,m,w$，表示有$n$朵云，$m$个搭配和你现有的钱的数目。第二行至$n+1$行，每行有两个整数，$c_i,d_i$，表示第$i$朵云的价钱和价值。第$n+2$至$n+1+m$行，每行有两个整数$u_i,v_i$。表示买第$u_i$朵云就必须买第$v_i$朵云，同理，如果买第$v_i$朵就必须买第$u_i$朵。

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

数据范围：
对于$30\%$的数据，满足$1\le n\le 100$；
对于$50\%$的数据，满足$1\le n,w\le 10^3$，$1\le m\le 100$；
对于$100\%$的数据，满足$1\le n\le 10^4$，$0\le m\le 5\times 10^3$。

思路是维护一个并查集，该并查集还需要维护每个集合的总成本和总价值，接下来就可以把每个集合视为一个物品来做$0-1$背包问题了。代码如下：

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

const int N = 1e4 + 10;
int n, m, vol;
int v[N], w[N];
int p[N];
int f[N];

int find(int x) {
    if (p[x] != 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++) scanf("%d%d", &v[i], &w[i]);
    for (int i = 1; i <= n; i++) p[i] = i;

    for (int i = 1; i <= m; i++) {
        int a, b;
        scanf("%d%d", &a, &b);
        int pa = find(a), pb = find(b);
        if (pa != pb) {
            v[pb] += v[pa];
            w[pb] += w[pa];
            p[pa] = pb;
        }
    }

    for (int i = 1; i <= n; i++) 
        if (p[i] == i) 
            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+nV)$，$V$是总容量（其实就是预算数），空间$O(n+V)$。