HDU 5313 bitset优化背包

时间:2023-03-09 19:07:03
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<> 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= ; i<l ; i++){
int v = vec[u][i];
if(col[v]>=) continue;
col[v] = col[u]^;
dfs(v , id);
}
} void solve()
{
memset(b, ,sizeof(b));
memset(w , , sizeof(w));
memset(col , - , sizeof(col));
id = ;
for(int i= ; i<=n ; i++){
if(col[i]<){
++id;
col[i] = ;
dfs(i , id);
}
}
} int cal()
{
bt[] = ;
for(int i= ; i<=id ; i++){
bt = bt|(bt<<b[i])|(bt<<w[i]);
}
int ans = n , ave = n/;
for(int i= ; 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= ; i<=n ; i++)vec[i].clear();
for(int i= ; i<m ; i++){
scanf("%d%d" , &u , &v);
vec[u].push_back(v);
vec[v].push_back(u);
}
solve();
printf("%d\n" , cal());
}
return ;
}