题目描述
两个整数A,B,如果他们某⼀数字相同了,那么(A,B)就是⼀组合法的数对(没有顺序),现在给定了N个整数,问存在多少对合法的数对呢?
输入
第⼀⾏,⼀个整数N。
接下来N⾏,每⾏⼀个正整数。
输出
输出⼀个整数,表示合法数对个数。
样例输入
复制样例数据
3
12
1
2
样例输出
2
提示
对于100%的数据,N≤1000000,每个正整数≤1018。
题解:对于一个数来说存在几个1和存在1个1,对答案的贡献都是一样的。
我们可以把正整数看数中出现了哪些0-9的数字。用二进制表示,最大的话也就是(1<<10)。然后我们处理好1-1023这些数字出现的次数。
如果a==b的话,对答案的贡献为n*(n-1)
如果a&b!=0,说明出现了相等的情况。那么对答案的贡献为a出现的次数*b出现的次数。
最后答案/2,因为多算了一遍
#include<bits/stdc++.h>
using namespace std;
const int N=3005;
typedef long long ll;
ll b[N];
int main()
{
int n;
ll x;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>x;
ll p=0;
while(x)
{
p|=1<<(x%10);
x/=10;
} //把这些压制二进制。
b[p]++;
}
ll ans=0;
int m=1<<10;
for(int i=1;i<=m;i++)
for(int j=1;j<=m;j++)
{
if(i==j) ans+=b[i]*(b[i]-1);
else if(i&j) //出现有元素相等了
{
ans+=b[i]*b[j];
}
}
printf("%lld\n",ans/2); //因为多算了一遍
return 0;
}