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

最新推荐文章于 2021-10-11 08:17:45 发布

。十二。

最新推荐文章于 2021-10-11 08:17:45 发布

阅读量395

点赞数

本文链接：https://blog.csdn.net/kejin2019/article/details/104419526

版权

纪中专栏收录该内容

209 篇文章 1 订阅

订阅专栏

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

题目描述
在顺利完成家庭作业以后， $M i r k o$ 感到非常的厌倦。所以，他列出了 $N$ 个数，这些数中有些数对他是喜欢的，有些数对他是不喜欢的。
他喜欢的数对叫做友好数对，如果两个数至少有一个相同的数字(不要求在相同的位置），那么这两个数就是友好数对。请帮助 $M i r k o$ 在这 $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);
}