搭配购买(DFS/并查集+01背包)

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

思路分析

题意大概就是要买一些东西，但是有一些条件，就是有些东西是有联系的，买这个必须买那个。所以先通过并查集或者DFS把这些有联系的先放到一起，转化成一个物品，然后再01背包就行了。

AC代码

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

const int N=10010,M=5010*2;

int n,m,w;
int c[N],d[N];
int u,v;
int e[M],ne[M],hd[N],idx;
int vis[N];
int tot=0,res=0,ans=0;
int T[N],R[N],idt,idr;
int dp[N];

void add(int x,int y){
	e[idx]=y;
	ne[idx]=hd[x];
	hd[x]=idx++;
}

void dfs(int x){
//	cout<<x<<" ";
	vis[x]=1;
	tot+=c[x];
	res+=d[x];
	
	for(int i=hd[x];i!=-1;i=ne[i]){
		int j=e[i];
		if(!vis[j]){
			dfs(j);
		}
	}
}

int main(){
	memset(hd,-1,sizeof hd); 
	scanf("%d %d %d",&n,&m,&w);
	for(int i=1;i<=n;i++){
		scanf("%d %d",&c[i],&d[i]);
	}	
	for(int i=1;i<=m;i++){
		scanf("%d %d",&u,&v);
		add(u,v),add(v,u);
	}
	
	for(int i=1;i<=n;i++){
		if(!vis[i]){
			tot=0,res=0;
			dfs(i);
//			cout<<endl;
//			cout<<"tot=="<<tot<<endl;
//			cout<<"res=="<<res<<endl;
			T[++idt]=tot;
			R[++idr]=res;
		}
	}
	
//	for(int i=1;i<=idt;i++){
//		printf("%d %d\n",T[i],R[i]);
//	}
	
	int len=idt;
	for(int i=1;i<=len;i++){
		for(int j=w;j>=T[i];j--){
			dp[j]=max(dp[j],dp[j-T[i]]+R[i]);
		}
	}
	
	printf("%d",dp[w]);
	
	return 0;
}