Codeforces Round #824 D. Meta-set

题意: 一张卡片有k种属性，属性值有3种可能的取值，分别是0/1/2。如果三张卡片的某一种属性的值两两不同或全都相同，那么这个属性就是好的。我们可以把三张卡片称为一个好集合，如果所有属性都是好的。一个由5张卡片构成的五元组可以被称为元集合，当且仅当你能从这5张卡片里面找出至少2个好集合。

        现在给你n张不同的卡片，问你能从这n张卡里找出多少个元集合。

一个重要的观察是：我任选两张卡片，能与这两张卡片构成好集合的第三张卡片是唯一确定的。

为什么呢？考虑每一种属性，如果这两张卡的这一种属性不同，那么第三张卡片的这一属性只能是与他们两个都不同的数。如果属性相同，那么第三张卡片的属性只能是与他们两个相同的数。我们可以顺理成章的确定出第三张卡片的所有属性。

另一个重要的观察是：一个五元组里面最多找出2个好集合。

c1,c2,c3,c4,c5表示一个五元组，假设c1,c2,c3是一个好集合，另一个好集合只能是c4,c5加上c1,c2,c3中的某一张。因为c1,c2,c3对另一个好集合只能贡献一张，你无法找到贡献两张能得到的另一个好集合，比如说，贡献c1,c2。根据第一个观察，跟c1,c2构成好集合的卡唯一确定，所以只能是c3。

根据上面的观察我们可以知道，一个元集合里面有一张共用卡，同属于两个好集合。

依据中间卡进行分类，元集合总数 = 以i号卡为中间卡的元集合数量（1 <= i <= n）

思路：题目条件中n最多是1000，暗示我们可以n^2去枚举两张卡，基于这两张卡推出第三张卡。

用一个map记录一下这种卡属于多少个好集合中。

最后枚举共用卡，如果这张卡属于x个好集合中，给答案加上x*(x-1)/2。

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
map<string,int> mp; 
signed main() {
	int n,k;
	cin>>n>>k;
	vector<array<int,20>> v(n+1);
	for(int i=1;i<=n;i++){
		array<int,20> a;
		for(int j=0;j<k;j++){
			cin>>a[j];
		}
		v[i]=a;
	}
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++){
			string p;
			auto v1=v[i],v2=v[j];
			for(int k=0;k<20;k++){
				if(v1[k]==v2[k]) p+=v1[k]+'0';
				else {
					int x=3-v1[k]-v2[k];
					p+=x+'0';
				}
			}
			mp[p]++;
		}
	}
	int ans=0;
	for(int i=1;i<=n;i++){
		string p;
		for(int j=0;j<20;j++){
			p+=v[i][j]+'0';
		}
		int x=mp[p];
		ans+=x*(x-1)/2;
	}
	cout<<ans<<endl;
}