洛谷 P7617 [COCI2011-2012#2] KOMPICI

洛谷 P7617 [COCI2011-2012#2] KOMPICI

题目描述

给定 $N$ 个正整数 $A_1,A_2,...,A_N$，求有多少整数对 $(i,j)$，满足以下条件：

• $1\le i<j\le N$
• $A_i$ 和 $A_j$ 至少有一位数字是相同的（不一定要在相同的数位）。

输入格式

输入的第一行包含一个正整数 $N$。

接下来 $N$ 行，每行包含一个正整数 $A_i$。

输出格式

输出一行一个整数，表示满足条件的整数对。

样例 #1

样例输入 #1

3
4
20
44

样例输出 #1

1

样例 #2

样例输入 #2

4
32
51
123
282

样例输出 #2

4

提示

【样例解释】

样例 1 中，满足要求的整数对为 $(1,3)$。

样例 2 中，满足要求的整数对为 $(1,3)$，$(1,4)$，$(2,3)$，$(3,4)$。

【数据范围】

对于 $100\%$ 的数据，$1\le N\le 10^6$，$1\le A_i\le 10^{18}$。

先说几句

难度低了，至少也得是绿题。

绝对是一道状态压缩的好题，带上容斥原理，怎是一道黄题？

解题思路

对于输入的每个数，统计 $0\cdots 9$ 每个数的出现。如果出现了，相应的状态位设为 $1$。在对于每种不同的状态进行统计，存入数组 $a$ 中。

记录数组 $b$，统计对于每个状态位有多少个状态位包含它。
枚举状态位 $i,j$，双重循环。如果状态 $i$ 完全包含于 $j$（$j$ 完全包含 $i$），就对 $b_i$ 累加 $a_j$。

计算答案时，枚举每个状态位，统计其中 $1$ 的个数。对于每种组合 $C_{b_x}^2$ 按照容斥原理，奇加偶减。

辅助解释

至于为什么这样做是对的？思考一下，其实，$b$ 中记录的实际上是什么？

是记录每种数位的组合情况被哪些数包含了！以此为基础，对于任意一个数对，已按照容斥原理计算出了。

奉上代码

// P7617 [COCI2011-2012#2] KOMPICI
#include <bits/stdc++.h>
#define int long long
#define SIZE 1024
#define all(x) x.begin(), x.end()
#define debug(x) cout<<#x<<":"<<x<<endl; 
using namespace std;

int a[SIZE]={0}, b[SIZE]={0};

signed main()
{
	int n; cin>>n;
	for(int i=0; i<n; i++)
	{
		int x; cin>>x;
		int y=0;
		while(x)
		{
			y|=1<<(x%10); x/=10;
		}
		a[y]++;
	}
	for(int i=0; i<SIZE; i++)
		for(int j=i; j<SIZE; j++)
			if((i|j)==j)		// j contains i or i is the base of j
				b[i]+=a[j];
	int sum=0;
	for(int x=1; x<SIZE; x++)
	{
		int onecount=0;			// count 1's
		for(int k=0; k<10; k++)
		{
			int t=1<<k; if(t&x) onecount++;
		}
		// the inclusion exclusion principle
		// odd: plus even: minus
		if(onecount%2==1) sum+=b[x]*(b[x]-1)/2;
		else              sum-=b[x]*(b[x]-1)/2;
	}
	cout<<sum;

    return 0;
}