背包+并查集 选学霸（洛谷 P2170）

选学霸

题目描述

老师想从N名学生中选M人当学霸，但有K对人实力相当，如果实力相当的人中，一部分被选上，另一部分没有，同学们就会抗议。所以老师想请你帮他求出他该选多少学霸，才能既不让同学们抗议，又与原来的M尽可能接近

输入格式

第一行，三个正整数N，M，K。

第2…K行,每行2个数，表示一对实力相当的人的编号(编号为1…N)

输出格式

一行，表示既不让同学们抗议，又与原来的M尽可能接近的选出学霸的数目。(如果有两种方案与M的差的绝对值相等，选较小的一种:)

---

很容易看出来是01背包加并查集的题目，但是难就难在这个01背包身上，这个背包跟普通的01背包不同，它可以超过背包容量，题目要求最接近背包容量的背包所装的价值。。。而不是不能超过背包容量。所以我们可以把背包容量开大两倍，然后之后找最接近的容量；这题有个坑点，当m=0时要特判；

代码：

#include<bits/stdc++.h>
using namespace std;
int fa[20100];
int find(int p){
	if(p==fa[p]) return p;
	return fa[p]=find(fa[p]);
}
int s[20010];//为每个队的人数 
int ss[20010];
int dp[40010];
int main(){
	int n,m,k;
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;i++) fa[i]=i;
	while(k--){
		int x,y;
		scanf("%d%d",&x,&y);
		fa[find(y)]=find(x);
	}
	for(int i=1;i<=n;i++) s[find(i)]++;
	int ans=0;//队的序号，数量 
	for(int i=1;i<=n;i++){
		if(s[i]) ss[++ans]=s[i];//每个队的人数 
	}
	for(int i=1;i<=ans;i++){
		for(int j=2*m;j>=1;j--){//不是m,保证队伍全部存入 
			if(j>=ss[i]) dp[j]=max(dp[j],dp[j-ss[i]]+ss[i]);
		}
	}
	int minn=2e9;
	int sum=2e9;
	for(int i=1;i<=2*m;i++){//找到最接近m的 
		//cout<<dp[i]<<endl;
		if(minn>abs(dp[i]-m)){
			minn=abs(dp[i]-m);
			sum=dp[i];
		} 
	}
	if(m) printf("%d\n",sum);
	else printf("0\n");
	return 0;
}