#include <bits/stdc++.h>
// solved by bekagg
#define ll long long
#define ent '\n'
#define pb push_back
#define all(x) x.begin(),x.end()
#define PRaim_bek_abi ios_base::sync_with_stdio(0);cin.tie(0);
using namespace std;
const int N = 1e5+5;
const int MOD = 1e9+7;
int n , d , c[5][5] , f[5];
ll ans;
map<string , int>mp;
void arkanefury228(){
cin >> n >> d;
if (d == 3){
while(true){
cout << 1;
}
return;
}
if (d == 0){
for (int i = 1; i <= n; i++){
string s;
cin >> s;
ans += mp[s];
mp[s]++;
}
cout << ans;
return;
}
if (d == 4) ans = n * (n - 1) / 2;
for (int i = 1; i <= n; i++){
string s;
cin >> s;
for (int mask = 0; mask < 16; mask++){
string g = "";
int sum = 0 , cnt = 0;
for (int j = 0; j < 4; j++){
if (mask & (1 << j)) {
g += s[j];
cnt++;
}
else {
g += 'X';
}
}
if (d == 4){
if (cnt == 0) continue;
if (cnt % 2 == 1){
ans -= mp[g];
}
else {
ans += mp[g];
}
continue;
}
if (d == 3){
if (cnt == 1){
ans += mp[g];
}
else if (cnt == 2){
ans -= mp[g] * 2;
}
else if (cnt == 3){
ans += mp[g] * 3;
}
else if (cnt == 4){
ans -= mp[g];
}
continue;
}
if (d == 2){
if (cnt == 0) continue;
if (cnt == 2){
ans += mp[g];
}
else if (cnt == 3){
ans -= mp[g] * 3;
}
else if (cnt == 4){
ans += mp[g] * 3;
}
}
if (d == 1){
if (cnt == 3){
ans += mp[g];
}
else if (cnt == 4){
ans -= mp[g] * 3;
}
}
}
for (int mask = 0; mask < 16; mask++){
string g = "";
for (int j = 0; j < 4; j++){
if (mask & (1 << j)) g += s[j];
else g += 'X';
}
mp[g]++;
}
}
cout << ans;
return;
}
signed main(){
PRaim_bek_abi
int t=1;
//cin>>t;
for (int respagold = 1; respagold <= t; respagold++) arkanefury228();
}
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
