Submission #1322923

#TimeUsernameProblemLanguageResultExecution timeMemory
1322923yerkoshPIN (CEOI10_pin)C++20
10 / 100
1096 ms9316 KiB
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ld long double
#define pb push_back
#define inf 1e18+1e9
#define F first
#define S second
#define int long long
#define mod (int)(1e9+7)
#define maxn 1000100
#define pii pair<int,int>
#define mod2 998244353
const int maxtime=1e8;
const long double eps=1e-9;
ld dist(ld x1,ld y1,ld x2,ld y2) {
    return sqrt(abs(x1-x2)*abs(x1-x2)+abs(y1-y2)*abs(y1-y2));
}


int n,d;
string s[50500];
int b[10];



void solve () {
    cin>>n>>d;
    for (int i=1;i<=n;i++) {
        cin>>s[i];
    }
    if (d==1) {
        map<string,int>mp1,mp2,mp3,mp4;
        int ans=0;
        for (int i=1;i<=n;i++) {
            ans+=mp1[s[i].substr(1)];
            mp1[s[i].substr(1)]++;
            ans+=mp2[s[i].substr(0,3)];
            mp2[s[i].substr(0,3)]++;
            string t=s[i].substr(1,1);
            t+=s[i].substr(2,2);
            ans+=mp3[t];
            mp3[t]++;
            t=s[i].substr(1,2);
            t+=s[i].substr(3,1);
            ans+=mp4[t];
            mp4[t]++;
        }
        cout<<ans;
        return;
    }
    for (int i=1;i<=n;i++) {
        for (int j=i+1;j<=n;j++) {
            int cnt=0;
            for (int k=0;k<4;k++) {
                if (s[i][k]!=s[j][k])cnt++;
            }
            b[cnt]++;
        }
    }
    cout<<b[d];
}
int32_t main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    int t=1;
    // cin>>t;
    for (int ii=1;ii<=t;ii++) {
        solve();
        // asd();
        if (ii<t)cout<<"\n";
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...