P7617 [COCI2011-2012#2] KOMPIĆI

恶心的二进制

一.原题地址

Link

题目

二.题目描述

给定 $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}$。

【说明】

本题分值按 COCI 原题设置，满分 $120$。

题目译自 COCI2011-2012 CONTEST #2 T4 KOMPIĆI。

三.思路

用桶来记录每个数的二进制
不然会TLE

四.AC代码

#include <iostream>

using namespace std;

const int N = 1e5 + 10;
long long ans;
int tmp[N], n;

int main() {
    cin >> n;
	while (n--) {
		long long a; 
        int x = 0;
		cin >> a;
        while(a) {
            x |= (1 << (a % 10));
        	a /= 10;
        }
		++tmp[x];
	}
	for (int i = 0; i < 1024; i++) {
		for (int j = i + 1; j < 1024; j++) {
			if (i & j) ans += (long long)tmp[i] * tmp[j];
        }
    }
	for (int i = 0; i < 1024; i++) {
		ans += (long long)tmp[i] * (tmp[i] - 1) / 2;
    }
	cout << ans << endl;
	return 0;
}