JZYZOJ1376 [coci2011]友好数对 容斥定理 状态压缩

http://172.20.6.3/Problem_Show.asp?id=1376

题意：找给出的数中含有相同数字的数对的对数。

mmp数论题竟然卡快读，莫名拉低通过率什么的太过分了。

刚开始想到了怎么容斥但是没法实现，看了标程发现需要状压，我还是太菜了。

代码

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm>  
#include<cmath>
using namespace std;
const long long maxn=1<<11;
const int modn=10000;
long long n;
long long a[maxn+10]={};
long long b[maxn+10]={};
long long read(){
    long long x=0;char ch=getchar();
    while(ch>'9'||ch<'0')ch=getchar();
    while(ch<='9'&&ch>='0'){x*=10;x+=ch-'0';ch=getchar();}
    return x;
}
int main(){
    n=read();
    long long x,y=0;
    int ma=1<<10;
    for(int i=1;i<=n;i++){
        x=read();;y=0;
        while(x){
            y|=1<<(x%10);x/=10;
        }
        a[y]++;
    }
    for(int i=1;i<ma;i++){
        for(int j=i;j<ma;j++){
            if((i|j)==j)b[i]+=a[j];
        }
    }
    long long ans=0;
    for(int i=1;i<ma;i++){
        if(b[i]==0||b[i]==1)continue;
        long long cnt=0;
        for(int j=0;j<10;j++){
            long long w=1<<j;w&=i;
            if(w)cnt++;
        }
        if(cnt&1)ans+=b[i]*(b[i]-1)/2;
        else ans-=b[i]*(b[i]-1)/2;
    }
    printf("%I64d
",ans);
    return 0;
}