洛谷 P2170 选学霸

来源:互联网 发布:kyocera打印机驱动mac 编辑:程序博客网 时间:2024/06/16 01:20

题目描述
老师想从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 20005using 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 重复用了,省内存,看不懂的新开一个数组就好。

原创粉丝点击