P1455 搭配购买

题目描述

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

输入格式

第一行输入三个整数，n,m,w，表示有 n 朵云，m 个搭配和你现有的钱的数目。

第二行至 n+1 行，每行有两个整数， ci​,di​，表示第 i 朵云的价钱和价值。

第 n+2 至 n+1+m 行 ，每行有两个整数 ui​,vi​。表示买第 ui​ 朵云就必须买第 vi​ 朵云,同理，如果买第 vi​ 朵就必须买第 ui​ 朵。

输出格式

一行，表示可以获得的最大价值。

输入输出样例

输入 #1

5 3 10
3 10
3 10
3 10
5 100
10 1
1 3
3 2
4 2

输出 #1

1

说明/提示

• 对于30% 的数据，满足 1≤n≤100；
• 对于 50% 的数据，满足 1≤n,w≤10^3，1≤m≤100；
• 对于 100% 的数据，满足 1≤n≤10^4，0≤m≤5×10^3。

分析

本题是并查集加01背包

AC代码

#include <iostream>
#include <cstdio>

using namespace std;

const int N = 1e4 + 100;
int fa[N], p[N], v[N], dp[N];

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

void merge(int x, int y) {
	int fx = find(x);
	int fy = find(y);
	if (fx != fy) {
		fa[fy] = fx;
		p[fx] += p[fy];
		v[fx] += v[fy];
	}
}

int main() {
	int n, m, w, x, y, maxn = 0;
	cin >> n >> m >> w;
	for (int i = 1; i <= n; i++) {
		fa[i] = i;
	}
	for (int i = 1; i <= n; i++) {
		scanf("%d%d", &p[i], &v[i]);
	}
	for (int i = 1; i <= m; i++) {
		cin >> x >> y;
		merge(x, y);
	}
	for(int i = 1; i <= n; i++) {
		if (i == find(i)) {
			for (int j = w; j >= p[i]; j--){
				dp[j] = max(dp[j], dp[j - p[i]] + v[i]);
				maxn = max(maxn, dp[j]);
			}
		}
	}
	cout << maxn << endl;
	return 0;
}