（纪中）1671. 友好数对（kompici）【位运算+容斥】

(File IO): input:kompici.in output:kompici.out
时间限制: 1000 ms 空间限制: 262144 KB 具体限制
Goto ProblemSet

---

题目描述
在顺利完成家庭作业以后，$Mirko$感到非常的厌倦。所以，他列出了$N$个数，这些数中有些数对他是喜欢的，有些数对他是不喜欢的。
他喜欢的数对叫做友好数对，如果两个数至少有一个相同的数字(不要求在相同的位置），那么这两个数就是友好数对。请帮助$Mirko$在这$N$个数找出有多少友好数对。

---

输入
第一行一个正整数$N（1<=N<=1000000）$。
接下来$N$行，每行一个正整数,范围在$1$到$1018$之间。$N$个数中任意两个数都是不同的。

输出
只有一行一个整数，表示友好数对的个数。

---

样例输入输出

---

数据范围限制
无

提示
无

---

解题思路
容斥原理
先把数拆分，用二进制表示
再用容斥原理求解

---

代码

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#include<iomanip>
#include<cmath>
using namespace std;
long long n,b[1000010],k,ans,s,x;
int main()
{
	freopen("kompici.in","r",stdin);
    freopen("kompici.out","w",stdout);
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%lld",&x);
        s=0;
        while(x>0)
        {
            k=x%10;
            x=x/10;
            s=s|(1<<k);
        }
        b[s]++;
    }
    for(int i=1; i<=1023; i++)
    {
    	if(b[i]==0) 
			continue;
        for(int j=1; j<=1023; j++)
        {
            if(!(i&j)||b[j]==0||i==j)
            	continue;
            ans=ans+b[i]*b[j];
        }
         ans=ans+b[i]*(b[i]-1);
    }
    printf("%lld",ans/2);
}