【容斥】Topcoder SRM div1-3 12004. SetAndSet

取反，变成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;
    }
};