P2170 选学霸

P2170 选学霸

题目描述
老师想从N名学生中选M人当学霸，但有K对人实力相当，如果实力相当的人中，一部分被选上，另一部分没有，同学们就会抗议。所以老师想请你帮他求出他该选多少学霸，才能既不让同学们抗议，又与原来的M尽可能接近 (如果有两种方案与M的差的绝对值相等，选较小的一种:)

---

调试日志： 背包内层循环变量写成了k

---

Solution

选能力相同的你就得把能力为这个的全部选上， 求最终选多少人和原定选择人数尽可能接近
先并查集一下， 每个块维护一个大小
现在一整块一整块的选就好了
可行性背包搞一下即可
最后从中间开始往两边枚举可行的就得

Code

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<climits>
#define LL long long
#define REP(i, x, y) for(int i = (x);i <= (y);i++)
using namespace std;
int RD(){
    int out = 0,flag = 1;char c = getchar();
    while(c < '0' || c >'9'){if(c == '-')flag = -1;c = getchar();}
    while(c >= '0' && c <= '9'){out = out * 10 + c - '0';c = getchar();}
    return flag * out;
    }
const int maxn = 80019;
int num, nr, k;
int father[maxn], size[maxn];
int findfather(int v){
    if(father[v] == v)return v;
    return father[v] = findfather(father[v]);
    }
void Union(int a, int b){
    int faA = findfather(a), faB = findfather(b);
    if(faA != faB){
        size[faB] += size[faA];
        father[faA] = faB;
        }
    }
int w[maxn], cnt;
int dp[maxn];
int main(){
    num = RD(), k = RD(), nr = RD();
    REP(i, 1, num)father[i] = i, size[i] = 1;
    REP(i, 1, nr){
        int x = RD(), y = RD();
        Union(x, y);
        }
    dp[0] = 1;
    REP(i, 1, num)if(father[i] == i)w[++cnt] = size[i];
    REP(i, 1, cnt){
        for(int j = k * 2;j >= w[i];j--){
            dp[j] |= dp[j - w[i]];
            }
        }
    REP(i, 0, k){
        if(dp[k - i]){printf("%d\n", k - i);return 0;}
        else if(dp[k + i]){printf("%d\n", k + i);return 0;}
        }
    }