B - Hyperset

Codeforces Round #612 (Div. 2) A~E2 题解

如果直接枚举的话,复杂度是\(O(n^3)\),显而易见会超时,但是我们会发现一个道理,当其中两个确定的时候,另外一个也就已经确定下来了,而我们要做的工作只是去寻找有没有这样一个东西,与其遍历每一个字符串去检查其是否符合条件,不如直接得到这个字符串,看是否存在这样一个字符串即可,这样就把复杂度将为了\(O(n^2)\)。妙啊

代码:

// Created by CAD on 2020/1/7.
#include <bits/stdc++.h>
using namespace std;

string s[1505];
map<string,int> m;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,k;cin>>n>>k;
    for(int i=1;i<=n;++i)
        cin>>s[i],m[s[i]]++;
    int all='S'+'T'+'E';
    int ans=0;
    for(int i=1;i<=n;++i)
        for(int j=i+1;j<=n;++j)
        {
            string t;
            for(int h=0;h<k;h++)
                if(s[i][h]==s[j][h])
                    t+=s[i][h];
                else t+=all-s[i][h]-s[j][h];
            ans+=m[t];
        }
    cout<<ans/3<<endl;
    return 0;
}