Submission #597544

#TimeUsernameProblemLanguageResultExecution timeMemory
597544Valaki2Cubeword (CEOI19_cubeword)C++14
84 / 100
240 ms31180 KiB
#include <bits/stdc++.h> using namespace std; #pragma GCC optimize("Ofast") //#pragma GCC optimize("unroll-loops") #pragma GCC target("avx2") #define int long long #define pb push_back #define mp make_pair #define pii pair<int, int> #define fi first #define se second const int mod = 998244353; const int maxn = 2e5; const int min_l = 3; const int max_l = 10; const int max_ch = 32; int combinations[max_l + 1][max_ch][max_ch]; int convert(char ch) { if(ch >= 'a' && ch <= 'p') { return ch - 'a'; } if(ch >= 'q' && ch <= 'z') { return ch - 'q' + 32; } if(ch >= 'A' && ch <= 'P') { return ch - 'A' + 16; } if(ch >= 'Q' && ch <= 'Z') { return ch - 'Q' + 42; } if(ch >= '0' && ch <= '9') { return ch - '0' + 52; } } int n; vector<string> v; int ans; int len; vector<int> corners; // vector<pii > neighbours; int number[max_ch][max_ch][max_ch]; int prec[max_ch][max_ch][max_ch][max_ch]; int try_scenario() { int res = 1; for(int i = 1; i <= 3; i++) { res = (res * combinations[len][corners[0]][corners[i]]) % mod; } return res; } void backtrack(int corner_id) { if(corner_id == 4) { number[corners[1]][corners[2]][corners[3]] = (number[corners[1]][corners[2]][corners[3]] + try_scenario()) % mod; } else { for(corners[corner_id] = 0; corners[corner_id] < max_ch; corners[corner_id]++) { backtrack(corner_id + 1); } } } void solve() { cin >> n; //n = 1; v.assign(2 * n, ""); for(int i = 0; i < n; i++) { cin >> v[i]; //v[i] = "cabac"; v[i + n] = v[i]; reverse(v[i + n].begin(), v[i + n].end()); } sort(v.begin(), v.end()); v.resize(unique(v.begin(), v.end()) - v.begin()); n = (int) v.size(); for(int i = 0; i < n; i++) { int cur_length = (int) v[i].size(); int cur_first = convert(v[i].front()); int cur_last = convert(v[i].back()); combinations[cur_length][cur_first][cur_last]++; } /*for(int i = 0; i < 8; i++) { for(int j = i + 1; j < 8; j++) { if(__builtin_popcount(i ^ j) == 1) { neighbours.pb(mp(i, j)); } } }*/ corners.assign(4, 0); for(len = min_l; len <= max_l; len++) { // reset everything for(int i = 0; i < max_ch; i++) { for(int j = 0; j < max_ch; j++) { for(int k = 0; k < max_ch; k++) { number[i][j][k] = 0; for(int l = 0; l < max_ch; l++) { prec[i][j][k][l] = combinations[len][i][j] * combinations[len][j][k] % mod * combinations[len][k][l] % mod; } } } } backtrack(0); for(int a = 0; a < max_ch; a++) { for(int b = 0; b < max_ch; b++) { for(int c = 0; c < max_ch; c++) { for(int d = 0; d < max_ch; d++) { ans = (ans + number[a][b][c] % mod * number[a][b][d] % mod * number[a][c][d] % mod * number[b][c][d] % mod) % mod; } } } } } cout << ans << "\n"; } signed main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); solve(); return 0; }

Compilation message (stderr)

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