P1455 搭配购买

95 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

传送门

思路:

用并查集将相同的云朵化为一个集合,并将一个集合内的所有云朵看作一个整体,最后用01背包得到答案。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int c[10010],d[10010];
int f[10010];
int dp[10010];
int sumc[10010],sumd[10010];
int ansc[10010],ansd[10010];

int find(int x)
{
	return f[x]==x?x:f[x]=find(f[x]);
}
int 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];
	}
	while(m--)
	{
		int u,v;
		cin>>u>>v;
		f[find(u)]=find(v);
	}
	for(int i = 1; i <= n; i++)
	{
		sumc[f[find(i)]] += c[i];
		sumd[f[find(i)]] += d[i];
	}
	int cnt = 1;
	for(int i = 1; i <= n; i++)
	{
		if(sumc[i] > 0)
		{
			ansc[cnt] = sumc[i];
			ansd[cnt++] = sumd[i];
		}
	}
	for(int i = 1; i < cnt; i++)
	{
		for(int j = w; j >= ansc[i]; j--)
		{
			dp[j] = max(dp[j], dp[j-ansc[i]]+ansd[i]);
		}
	}
	cout<<dp[w]<<endl;
}