算法提高--数据结构--并查集例题(2)

算法提高–数据结构–并查集例题–购买搭配

题目描述：

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

输入格式：

第 1 行包含三个整数 n，m，w，表示有 n 朵云，m 个搭配，Joe有 w的钱。第 2∼n+1行，每行两个整数 ci，di 表示 i朵云的价钱和价值。第 n+2∼n+1+m行，每行两个整数 ui，vi，表示买 ui 就必须买 vi，同理，如果买 vi 就必须买 ui。

输出格式：

一个整数表示最大价值

数据范围：

1≤n≤10000,
0≤m≤5000,
1≤w≤10000,
1≤ci≤5000,
1≤di≤100,
1≤ui,vi≤n

示例：

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

1

解题思路：

很明显可以用并查集和背包来写，将搭配在一起的云放在一个集合，将集合里的所有云的价值和价格都加到代表元，接下就是背包问题了。

AC代码：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 1e4 + 10;

int f[N];
int p[N];
int n, m, v;
int w[N], c[N];

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

int main()
{
	scanf("%d%d%d", &n, &m, &v);
	
	for(int i = 1; i <= n; i ++ ) p[i] = i;
	
	for(int i = 1; i <= n; i ++ )
		scanf("%d%d", &c[i], &w[i]);
	
	while(m -- )
	{
		int a, b, pa, pb;
		scanf("%d%d", &a, &b);
		pa = find(a), pb = find(b);
		if(pa != pb)
		{
			p[pa] = pb;
			w[pb] += w[pa], c[pb] += c[pa]; 
		}
	}
	
	for(int i = 1; i <= n; i ++ )
	{
		if(p[i] != i) continue;
		for(int j = v; j >= c[i]; j -- )
			f[j] = max(f[j], f[j - c[i]] + w[i]);
	}
	
	cout << f[v] << endl;
	
	return 0;
}