题意: 一张卡片有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;
}