题目
题目描述
老师想从N名学生中选M人当学霸,但有K对人实力相当,如果实力相当的人中,一部分被选上,另一部分没有,同学们就会抗议。所以老师想请你帮他求出他该选多少学霸,才能既不让同学们抗议,又与原来的M尽可能接近
输入格式
第一行,三个正整数N,M,K。
第2…K行,每行2个数,表示一对实力相当的人的编号(编号为1…N)
输出格式
一行,表示既不让同学们抗议,又与原来的M尽可能接近的选出学霸的数目。(如果有两种方案与M的差的绝对值相等,选较小的一种:)
输入输出样例
输入 #1
4 3 2
1 2
3 4
输出 #1
2
说明/提示
100%的数据N,P<=20000
算法 01背包+并查集
时间复杂度 最坏O(n^2)
思路
题目中已经明确地说了,如果两人实力相当,只选其中一个,必定无法满足没有人有争议的条件,
那么就只好这些实力相当的人要么都不选,要么都选,
显然实力相当的人是在一个集合里的,要维护集合信息,很自然地就想到了并查集,
用并查集把这些人划分到一个集合里,然后计算出每一个集合的大小,
开一个数组把表示集合大小的这些数存在起来。
搞定好这些后,就用01背包,计算方案,
最后找出一个离m最近的方案,若有很多,输出最小的那一个。
代码
#include<bits/stdc++.h>
using namespace std;
int n,m,k,f[200005],d[200005],l[200000];
int get(int x){
if(f[x]==x) return x;
f[x]=get(f[x]);
return f[x];
}
void hb(int x,int y){
int t1=get(x),t2=get(y);
if(f[t1]!=f[t2]){
f[t1]=t2;
}
}
vector<int> v;
int main(){
cin>>n>>m>>k;
for(int i=1;i<=n;i++) f[i]=i;
for(int i=1;i<=k;i++){
int a,b;
cin>>a>>b;
hb(a,b);
}
for(int i=1;i<=n;i++){
int fa=get(i);
d[fa]++;//统计集和大小
}
for(int i=1;i<=n;i++){
if(d[i])
v.push_back(d[i]);
}
l[0]=1;
for(int i=0;i<v.size();i++)
for(int j=n;j>=v[i];j--)
if(l[j-v[i]])
l[j]=1;
int ans=999999999;
for(int i=0;i<=n;i++)
if(l[i]){
if(abs(ans-m)>abs(i-m))
ans=i;
}
cout<<ans;
return 0;
}