Codevs_P3372 选学霸(并查集+DP+背包)

时间限制: 1 s
空间限制: 128000 KB
题目等级 : 大师 Master
题目描述 Description
老师想从N名学生中选M人当学霸，但有K对人实力相当，如果实力相当的人中，一部分被选上，另一部分没有，同学们就会抗议。所以老师想请你帮他求出他该选多少学霸，才能既不让同学们抗议，又与原来的M尽可能接近。

输入描述 Input Description
第一行，三个正整数N，M，K。

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

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

样例输入 Sample Input
4 3 2
1 2
3 4

样例输出 Sample Output
2

数据范围及提示 Data Size & Hint
100%的数据N，P<=30000

#include<cstdio>
#include<map>
#include<iostream>
using namespace std;
#define N 30005
int n,m,k,t;int f[N],d[N];
bool p[N];map<int,int> cl;
int find(int x){
    if(f[x]!=x) return f[x]=find(f[x]);
    return f[x];
}
int main(){
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++) f[i]=i,d[i]=1;
    int x,y,r1,r2;
    while(k--){
        scanf("%d%d",&x,&y);
        r1=find(x),r2=find(y);
        if(r1!=r2) f[r1]=r2;
    }
    for(int i=1;i<=n;i++)   if(find(i)==i)t++,cl[i]=t;
    for(int i=1;i<=n;i++){
        r1=find(i);
        if(r1!=i){d[cl[r1]]++;}
    }
    p[0]=true;
    for(int i=1;i<=t;i++){
        for(int j=n;j>=d[i];j--){
            if(p[j-d[i]]) p[j]=true;
        }
    }
    int l=m,r=m;
    while(l>=0&&r<=n){
        if(p[l]||p[r]){
            if(p[l]) {printf("%d",l);return 0;}
            else {printf("%d",r);return 0;}
        }
        l--,r++;
    }
    return 0;
}