P1455 搭配购买 ——并查集+动态规划dp

搭配购买

题目描述

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

输入格式

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

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

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

输出格式

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

样例 #1

样例输入 #1

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

样例输出 #1

1

提示

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

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

using namespace std;
#define int long long
const int N=100010;

int f[N],c[N],d[N],dp[N];

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

signed main(){
int n,m,w;cin>>n>>m>>w;

for(int i=1;i<=n;i++)f[i]=i;

for(int i=1;i<=n;i++)cin>>c[i]>>d[i];

for(int i=1;i<=m;i++){
	int a,b;cin>>a>>b;
	int t1=find(a);
	int t2=find(b);
	if(t1!=t2){
		f[t2]=t1;
		c[t1]+=c[t2];
		c[t2]=0;
		d[t1]+=d[t2];
		d[t2]=0;
	} 
	
}

for(int i=1;i<=n;i++){
	for(int j=w;j>=c[i];j--){
		dp[j]=max(dp[j],dp[j-c[i]]+d[i]);
	}
}

cout<<dp[w];

} 

将同一个集合的价格和价值加在父节点上，再将子节点的价格和价值清零