取反,变成or和相同。同一位,为1的不能放在同一边,并查集加容斥搞。
第一次在TC上做题,机制轻喷。。。
代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
class SetAndSet{
public:
typedef long long LL;
int n,fa[56];
LL ans;
vector<int>b[26];
int get(int x){return fa[x]==x?x:fa[x]=get(fa[x]);}
void dfs(int x,int y){
if(x>=20){
int cnt=0;
for(int i=1;i<=n;i++)if(get(i)==i)cnt++;
if(y&1)ans-=(1LL<<cnt)-2;else ans+=(1LL<<cnt)-2;
return;
}
dfs(x+1,y);
if(b[x].size()<1)return;
int tmp[56],t=get(b[x][0]);
for(int i=1;i<=n;i++)tmp[i]=fa[i];
for(int i=1,tt;i<b[x].size();i++){tt=get(b[x][i]);if(tt!=t)fa[tt]=t;}
dfs(x+1,y+1);
for(int i=1;i<=n;i++)fa[i]=tmp[i];
}
LL countandset(vector<int> a){
n=a.size();ans=0;
for(int i=1;i<=n;i++)fa[i]=i;
for(int i=0;i<20;i++)b[i].clear();
for(int j=0;j<20;j++)
for(int i=1;i<=n;i++) if(~a[i-1]>>j&1)b[j].push_back(i);
dfs(0,0);
return ans;
}
};