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=<<;
 const int modn=;
 long long n;
 long long a[maxn+]={};
 long long b[maxn+]={};
 long long read(){
     long long x=;char ch=getchar();
     while(ch>''||ch<'')ch=getchar();
     while(ch<=''&&ch>=''){x*=;x+=ch-'';ch=getchar();}
     return x;
 }
 int main(){
     n=read();
     long long x,y=;
     int ma=<<;
     for(int i=;i<=n;i++){
         x=read();;y=;
         while(x){
             y|=<<(x%);x/=;
         }
         a[y]++;
     }
     for(int i=;i<ma;i++){
         for(int j=i;j<ma;j++){
             if((i|j)==j)b[i]+=a[j];
         }
     }
     long long ans=;
     for(int i=;i<ma;i++){
         if(b[i]==||b[i]==)continue;
         long long cnt=;
         for(int j=;j<;j++){
             long long w=<<j;w&=i;
             if(w)cnt++;
         }
         if(cnt&)ans+=b[i]*(b[i]-)/;
         else ans-=b[i]*(b[i]-)/;
     }
     printf("%I64d\n",ans);
     return ;
 }