Submission #788422

#TimeUsernameProblemLanguageResultExecution timeMemory
788422math_rabbit_1028PIN (CEOI10_pin)C++14
100 / 100
66 ms28120 KiB
#include <bits/stdc++.h> using namespace std; long long n, d, arr[36][36][36][36]; long long zero, one[36][4], two[36][36][6], three[36][36][36][4], four[36][36][36][36]; int CharToInt(char c) { if (c <= '9') return c - '0'; if (c >= 'a') return c - 'a' + 10; } int main() { ios_base::sync_with_stdio(false); cin.tie(0); cin >> n >> d; for (int i = 1; i <= n; i++) { char s[10]; cin >> s; arr[CharToInt(s[0])][CharToInt(s[1])][CharToInt(s[2])][CharToInt(s[3])] = 1; } for (int i = 0; i < 36; i++) { for (int j = 0; j < 36; j++) { for (int k = 0; k < 36; k++) { for (int l = 0; l < 36; l++) { four[i][j][k][l] = arr[i][j][k][l] * (arr[i][j][k][l] + 1) / 2; } } } } for (int i = 0; i < 36; i++) { for (int j = 0; j < 36; j++) { for (int k = 0; k < 36; k++) { for (int l = 0; l < 36; l++) { three[i][j][k][0] += arr[i][j][k][l]; three[i][j][k][1] += arr[i][j][l][k]; three[i][j][k][2] += arr[i][l][j][k]; three[i][j][k][3] += arr[l][i][j][k]; } three[i][j][k][0] = three[i][j][k][0] * (three[i][j][k][0] + 1) / 2; three[i][j][k][1] = three[i][j][k][1] * (three[i][j][k][1] + 1) / 2; three[i][j][k][2] = three[i][j][k][2] * (three[i][j][k][2] + 1) / 2; three[i][j][k][3] = three[i][j][k][3] * (three[i][j][k][3] + 1) / 2; for (int l = 0; l < 36; l++) { three[i][j][k][0] -= four[i][j][k][l]; three[i][j][k][1] -= four[i][j][l][k]; three[i][j][k][2] -= four[i][l][j][k]; three[i][j][k][3] -= four[l][i][j][k]; } } } } for (int i = 0; i < 36; i++) { for (int j = 0; j < 36; j++) { for (int k = 0; k < 36; k++) { for (int l = 0; l < 36; l++) { two[i][j][0] += arr[i][j][k][l]; two[i][j][1] += arr[i][k][j][l]; two[i][j][2] += arr[i][k][l][j]; two[i][j][3] += arr[k][i][j][l]; two[i][j][4] += arr[k][i][l][j]; two[i][j][5] += arr[k][l][i][j]; } } two[i][j][0] = two[i][j][0] * (two[i][j][0] + 1) / 2; two[i][j][1] = two[i][j][1] * (two[i][j][1] + 1) / 2; two[i][j][2] = two[i][j][2] * (two[i][j][2] + 1) / 2; two[i][j][3] = two[i][j][3] * (two[i][j][3] + 1) / 2; two[i][j][4] = two[i][j][4] * (two[i][j][4] + 1) / 2; two[i][j][5] = two[i][j][5] * (two[i][j][5] + 1) / 2; for (int k = 0; k < 36; k++) { two[i][j][0] -= three[i][j][k][0]; two[i][j][1] -= three[i][k][j][0]; two[i][j][2] -= three[i][k][j][1]; two[i][j][3] -= three[k][i][j][0]; two[i][j][4] -= three[k][i][j][1]; two[i][j][5] -= three[k][i][j][2]; } for (int l = 0; l < 36; l++) { two[i][j][0] -= three[i][j][l][1]; two[i][j][1] -= three[i][j][l][2]; two[i][j][2] -= three[i][l][j][2]; two[i][j][3] -= three[i][j][l][3]; two[i][j][4] -= three[i][l][j][3]; two[i][j][5] -= three[l][i][j][3]; } for (int k = 0; k < 36; k++) { for (int l = 0; l < 36; l++) { two[i][j][0] -= four[i][j][k][l]; two[i][j][1] -= four[i][k][j][l]; two[i][j][2] -= four[i][k][l][j]; two[i][j][3] -= four[k][i][j][l]; two[i][j][4] -= four[k][i][l][j]; two[i][j][5] -= four[k][l][i][j]; } } } } for (int i = 0; i < 36; i++) { for (int j = 0; j < 36; j++) { for (int k = 0; k < 36; k++) { for (int l = 0; l < 36; l++) { one[i][0] += arr[i][j][k][l]; one[i][1] += arr[j][i][k][l]; one[i][2] += arr[j][k][i][l]; one[i][3] += arr[j][k][l][i]; } } } one[i][0] = one[i][0] * (one[i][0] + 1) / 2; one[i][1] = one[i][1] * (one[i][1] + 1) / 2; one[i][2] = one[i][2] * (one[i][2] + 1) / 2; one[i][3] = one[i][3] * (one[i][3] + 1) / 2; for (int j = 0; j < 36; j++) { one[i][0] -= two[i][j][0]; one[i][1] -= two[j][i][0]; one[i][2] -= two[j][i][1]; one[i][3] -= two[j][i][2]; } for (int k = 0; k < 36; k++) { one[i][0] -= two[i][k][1]; one[i][1] -= two[i][k][3]; one[i][2] -= two[k][i][3]; one[i][3] -= two[k][i][4]; } for (int l = 0; l < 36; l++) { one[i][0] -= two[i][l][2]; one[i][1] -= two[i][l][4]; one[i][2] -= two[i][l][5]; one[i][3] -= two[l][i][5]; } for (int j = 0; j < 36; j++) { for (int k = 0; k < 36; k++) { one[i][0] -= three[i][j][k][0]; one[i][1] -= three[j][i][k][0]; one[i][2] -= three[j][k][i][0]; one[i][3] -= three[j][k][i][1]; } } for (int j = 0; j < 36; j++) { for (int l = 0; l < 36; l++) { one[i][0] -= three[i][j][l][1]; one[i][1] -= three[j][i][l][1]; one[i][2] -= three[j][i][l][2]; one[i][3] -= three[j][l][i][2]; } } for (int k = 0; k < 36; k++) { for (int l = 0; l < 36; l++) { one[i][0] -= three[i][k][l][2]; one[i][1] -= three[i][k][l][3]; one[i][2] -= three[k][i][l][3]; one[i][3] -= three[k][l][i][3]; } } for (int j = 0; j < 36; j++) { for (int k = 0; k < 36; k++) { for (int l = 0; l < 36; l++) { one[i][0] -= four[i][j][k][l]; one[i][1] -= four[j][i][k][l]; one[i][2] -= four[j][k][i][l]; one[i][3] -= four[j][k][l][i]; } } } } long long three_tot = 0, two_tot = 0, one_tot = 0, zero_tot = n * (n - 1) / 2; for (int i = 0; i < 36; i++) { for (int j = 0; j < 36; j++) { for (int k = 0; k < 36; k++) { three_tot += three[i][j][k][0]; three_tot += three[i][j][k][1]; three_tot += three[i][j][k][2]; three_tot += three[i][j][k][3]; } } } for (int i = 0; i < 36; i++) { for (int j = 0; j < 36; j++) { two_tot += two[i][j][0]; two_tot += two[i][j][1]; two_tot += two[i][j][2]; two_tot += two[i][j][3]; two_tot += two[i][j][4]; two_tot += two[i][j][5]; } } for (int i = 0; i < 36; i++) { one_tot += one[i][0]; one_tot += one[i][1]; one_tot += one[i][2]; one_tot += one[i][3]; } zero_tot = zero_tot - one_tot - two_tot - three_tot; if (d == 1) cout << three_tot << "\n"; if (d == 2) cout << two_tot << "\n"; if (d == 3) cout << one_tot << "\n"; if (d == 4) cout << zero_tot << "\n"; return 0; }

Compilation message (stderr)

pin.cpp: In function 'int CharToInt(char)':
pin.cpp:9:1: warning: control reaches end of non-void function [-Wreturn-type]
    9 | }
      | ^
#Verdict Execution timeMemoryGrader output
Fetching results...