洛谷 P2170 选学霸

题目描述
老师想从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

//并查集搞一下 + DP  也可以Tarjan 
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define MAXN 20005
using namespace std;
int par[MAXN],rank[MAXN],siz[MAXN];
int dp[MAXN],n,m,k;

inline void read(int &x){
    x=0; int f=1; char c=getchar();
    while(c>'9'||c<'0'){ if(c=='-')f=-1; c=getchar(); }
    while(c>='0'&&c<='9'){ x=x*10+c-'0'; c=getchar(); } 
}

int Find(int x){
    if(x == par[x]) return x;
    else return par[x] = Find(par[x]);
}

inline void Megre(int x,int y){
    x = Find(x), y = Find(y);
    if(x == y) return ;
    if(rank[x] <  rank[y]) par[x] = y;
    else {
        if(rank[x] == rank[y]) ++rank[x];
        par[y] = x;
    }
}

int main(int argc,char *argv[]){
    read(n),read(m),read(k);
    for(int i=1; i<=n; ++i) par[i] = i;
    for(int u,v,i=1; i<=k; ++i){
        read(u),read(v);
        if(Find(u) != Find(v)) Megre(u,v);
    }
    for(int i=1; i<=n; ++i) ++siz[Find(i)];
    int Num = 0;
    for(int i=1; i<=n; ++i)
        if(siz[i]) siz[Num ++] = siz[i];
    for(int i=0; i<Num; ++i)
        for(int j=n; j>=siz[i]; --j)
            if( abs(m - dp[j]) > abs(m - dp[j - siz[i]] - siz[i]) )
                dp[j] = dp[j - siz[i]] + siz[i];
    int Ans = 99999999;
    for(int i=0; i<=n; ++i)
        if(abs(m - dp[i]) < abs(m - Ans)) Ans = dp[i];
    printf("%d\n" ,Ans);
    return 0;
}

siz 重复用了，省内存，看不懂的新开一个数组就好。