洛谷P2170 并查集+背包详细题解

题目：https://www.luogu.com.cn/problem/P2170

思路：看到这道题的第一反应就是01背包，即是我选or不选这个人的时候我最多选了多少人，然后寻找最接近m的即可。但题目加了一个限制，同等”聪明“的人得一起选，显然就是并查集啦，把同等聪明的人放入一个集合，加上01背包的选or不选这个集合，求出每个状态下的选人数即可。

以下是ac代码（附详细注释）

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define lop(i,a,b) for(int i=(a);i<(b);i++)
#define dwn(i,a,b) for(int i=(a);i>=(b);i--)
#define el '\n'
using LL = long long;
using db = double;
const int N = 2e4 + 10;
int n, m, k, idx = 1;
int f[N], a[N],flag[N];//老大，每个组别的人数量
int dp[N];//dp[i]表示还能选i个人时，选了多少人
int find(int x)//查集
{
	return f[x] == x ? x : f[x] = find(f[x]);
}
void judge()//放入一个集合的函数
{
	int x, y;
	scanf("%d%d", &x, &y);
	f[find(x)] = find(y);//并集
}
void solve()
{
	int ans = 1e8;
	dp[0] = 0;
	rep(i, 1, idx)
	{
		dwn(j, n, a[i])//选人最多是可以全选的，所以背包大小从n开始
			dp[j] = max(dp[j], dp[j - a[i]] + a[i]);
	}
	rep(i, 0, n)
		ans = min(abs(dp[i] - m), ans);//找出最接近m的位置
	rep(i, 0, n)
	{
		if (abs(dp[i] - m) == ans)
		{
			cout << dp[i] << el;//输出
			break;
		}
	}
}
int main()
{
	memset(a, 0, sizeof(a));
	memset(flag, 0, sizeof(flag));
	cin >> n >> m >> k;
	//集合数太多，会导致tle（比如k==0，所以进行一个特判）
	if (!k)
	{
		cout << min(m, n);//输出小的那个就行了
		return 0;
	}
	rep(i, 1, n)//并查集初始化
		f[i] = i;
	rep(i, 1, k)//并集
		judge();
	rep(i, 1, n - 1)
	{
		if (flag[i])//如果已经放到别的集合就跳过
			continue;
		rep(j, i + 1, n)
		{
			if (!flag[j] && find(i) == find(j))
			{
				a[idx]++;//放入一个集合，用数组表示，集合数就是人数
				flag[j] = 1;//标记已经进入集合了
			}
		}
		if(!flag[i])
			a[idx]++;
		flag[i] = 1;//i也入集合
		idx++;
	}
	if (!flag[n])//n特判
		a[idx] = 1;
	solve();
	return 0;
}