并查集-搭配购买

并查集-搭配购买

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

输入
第 1 行 n，m，w（1<n,m,w<=1000），表示 n 朵云，m 个搭配，Joe 有 w 的钱。 第 2~n+1 行，每行 ci，di 表示 i 朵云的价钱和价值。 第 n+2~n+1+m 行，每行 ui，vi，表示买 ui 就必须买 vi，同理，如果买 vi 就必须买 ui

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

样例输入

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

样例输出

1

提醒

• 最后返回的是拥有的钱可以购买的最多的价值，也就是说可以购买多个集合

代码

#include<iostream>
 
using namespace std;
 
const int N = 1010;

int fa[N];	// 每朵云的父亲 
int price[N];	//	每朵云的价格 
int value[N];	//	每朵云的价值 

int n,m,w;	// n是有几朵云 m是关系数 w是拥有的钱 

int f[N];	// 存储每个下标所对应的钱可以买到的最大的价值 

//	返回x最上面的父亲  并将x直接指向最高层的父亲 
int find(int x){
    if(fa[x] != x) fa[x] = find(fa[x]);
    return fa[x];
}
 
int main(){
	
    cin >> n >> m >> w;
    // 初始化每个点都是它自己的父亲 
	for(int i = 1; i <= n;i++){
        fa[i] = i;
    }
    //	输入 
    for(int i = 1; i <= n;i++){
        int p,v;
        cin >> p >> v;
        price[i] = p;
        value[i] = v;
    }
    //	输入关系 
    for(int i = 1; i <= m;i++){
        int a,b;
        cin >> a >> b;
        int fai = find(a);
        int fbi = find(b);
        //	当两个点的父亲都不一样的时候 合并集合 
        //	父亲节点存放整个集合的价格 和 价值 
		if(fai != fbi){
            fa[fai] = fbi;
            price[fbi] += price[fai];
            value[fbi] += value[fai]; 
        }
    }
    
    //找出0-w块钱每块钱能买到最多的价值	 
    for(int i = 1; i <= n;i++){
    	//	如果这个点是父亲节点 
        if(fa[i] == i){
        	//	循环这个节点总价格到拥有的钱 
        	for(int j = w; j >= price[i];j--){
        		//	j-price[i] 是 除去买这个集合还剩下的钱
				//	f[i-price[i]] 是除去这个集合的最高的价值 
        		f[j] = max(f[j],f[j-price[i]] + value[i]);
			}
		}
    }
    cout << f[w] <<endl;
}