搭配购买（并查集，01背包）

题目描述

n件物品，每件物品都有重量w和价值v，给出m个关系，每个关系用 (a，b)表示，表示a与b必须同时买。现在能承受wei的重量，问能获得的最大价值。

链接

---

分析

这其实并不是一个有依赖的分组背包，将有关系的一些物品看作一个物品就是01背包问题了。
用并查集合并，并且合并时更新一下每个集合的w与v即可。

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 10010;

int n, m, wei;
int w[N], v[N], fa[N], dp[N];

int get(int x)
{
    if (fa[x] == x) return x;
    return fa[x] = get(fa[x]);
}

int main()
{
    scanf("%d%d%d", &n, &m, &wei);
    for (int i = 1; i <= n; i++) fa[i] = i;
    for (int i = 1; i <= n; i++)
        scanf("%d%d", &w[i], &v[i]);
    for (int i = 1; i <= m; i++)
    {
        int x, y;
        scanf("%d%d", &x, &y);
        int rx = get(x), ry = get(y);
        if (rx == ry) continue;
        w[ry] += w[rx], v[ry] += v[rx];
        fa[rx] = ry;
    }
    memset(dp, -0x3f, sizeof(dp));	// 01背包有两种模型，这里用恰好装满的模型
    dp[0] = 0;
    for (int i = 1; i <= n; i++)
    {
        if (fa[i] != i) continue;
        for (int j = wei; j >= w[i]; j--)
            dp[j] = max(dp[j], dp[j - w[i]] + v[i]);
    }
    int ans = 0;
    for (int i = 0; i <= wei; i++) ans = max(ans, dp[i]);
    printf("%d\n", ans);
    return 0;
}