HDU 5313 bitset优化背包

 题目大意：

添加尽可能少的边，最后使图形成二分图

一开始将图区分成一个个联通分量，根据二分图染色，计算出每个联通分量的黑色点和白色点的个数

希望添加的边最少，那么合并的时候，希望黑白块尽可能平均，这无疑背包dp做，但超时了。。。T T

跟着题解说的bitset，学了一下，果然总共10000个点不到，那么只要用bitset上的某一位代表取到的值即可- -，好神奇。。这里用的是或运算

#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <vector>
#include <queue>
#include <climits>
#include <bitset>
#include <algorithm>
using namespace std;
#define ls o<<1
#define rs o<<1|1
#define define_m int m=(l+r)>>1
#define N 10010
#define ll long long
vector <int> vec[N];
bitset<10001> bt;
int n , m , u , v , id;
int w[N] , b[N] ,col[N];

void dfs(int u , int id)
{
    if(col[u]) w[id]++;
    else b[id]++;
    int l = vec[u].size();
    for(int i=0 ; i<l ; i++){
        int v = vec[u][i];
        if(col[v]>=0) continue;
        col[v] = col[u]^1;
        dfs(v , id);
    }
}

void solve()
{
    memset(b, 0,sizeof(b));
    memset(w , 0 , sizeof(w));
    memset(col , -1 , sizeof(col));
    id = 0;
    for(int i=1 ; i<=n ; i++){
        if(col[i]<0){
            ++id;
            col[i] = 0;
            dfs(i , id);
        }
    }
}

int cal()
{
    bt[0] = 1;
    for(int i=1 ; i<=id ; i++){
        bt = bt|(bt<<b[i])|(bt<<w[i]);
    }
    int ans = n , ave = n/2;
    for(int i=0 ; i<=n ; i++){
        if(bt[i]){
            if(abs(ans-ave)>abs(i-ave)) ans = i;
        }
    }
    return ans*(n-ans)-m;
}

int main()
{
    //freopen("in.txt" , "r" , stdin);
    int T;
    scanf("%d" , &T);
    while(T--)
    {
        scanf("%d%d" , &n , &m);
        for(int i=1 ; i<=n ; i++)vec[i].clear();
        for(int i=0 ; i<m ; i++){
            scanf("%d%d" , &u , &v);
            vec[u].push_back(v);
            vec[v].push_back(u);
        }
        solve();
        printf("%d
" , cal());
    }
    return 0;
}