Submission #948562

#	Time	Username	Problem	Language	Result	Execution time	Memory
948562		GrandTiger1729	Cubeword (CEOI19_cubeword)	C++17	84 / 100	1200 ms	9852 KiB

cubeword

#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;

namespace
{
const int N = 11, K = 128, MOD = 998244353;
int cnt[N][K][K]{}, res[K][K][K]{};
const string charset = "0123456789qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLZXCVBNM";
inline void addmod(int &x, int y)
{
    x += y;
    if (x > MOD)
    {
        x -= MOD;
    }
}
}
int main()
{
    cin.tie(0)->sync_with_stdio(0);
    int n;
    cin >> n;
    vector<string> a(n);
    for (int i = 0; i < n; i++)
    {
        string s;
        cin >> s;
        string ss = s;
        reverse(ss.begin(), ss.end());
        if (ss < s)
        {
            swap(s, ss);
        }
        a[i] = s;
    }
    sort(a.begin(), a.end());
    a.resize(unique(a.begin(), a.end()) - a.begin());
    n = a.size();
    for (int i = 0; i < n; i++)
    {
        string s = a[i], ss = s;
        reverse(ss.begin(), ss.end());
        if (s == ss)
        {
            cnt[s.size()][s[0]][ss[0]]++;
        }
        else
        {
            cnt[s.size()][s[0]][ss[0]]++;
            cnt[s.size()][ss[0]][s[0]]++;
        }
    }
    int ans = 0;
    for (int t = 3; t < N; t++)
    {
        for (int i : charset)
        {
            for (int j : charset)
            {
                for (int k : charset)
                {
                    res[i][j][k] = 0;
                }
            }
        }
        for (int l : charset)
        {
            for (int i : charset)
            {
                if (cnt[t][i][l] == 0)
                {
                    continue;
                }
                for (int j : charset)
                {
                    if (cnt[t][j][l] == 0)
                    {
                        continue;
                    }
                    for (int k : charset)
                    {
                        if (cnt[t][k][l] == 0)
                        {
                            continue;
                        }
                        // (res[i][j][k] += 1ll * cnt[t][i][l] * cnt[t][j][l] * cnt[t][k][l]) %= MOD;
                        addmod(res[i][j][k], 1ll * cnt[t][i][l] * cnt[t][j][l] * cnt[t][k][l] % MOD);
                    }
                }
            }
        }
        for (int i : charset)
        {
            for (int j : charset)
            {
                for (int k : charset)
                {
                    if (res[i][j][k] == 0)
                    {
                        continue;
                    }
                    for (int l : charset)
                    {
                        // (ans += res[i][j][k] * res[i][j][l] % MOD * res[i][k][l] % MOD * res[j][k][l]) %= MOD;
                        addmod(ans, (__int128_t)1 * res[i][j][k] * res[i][j][l] * res[i][k][l] * res[j][k][l] % MOD);
                    }
                }
            }
        }
    }
    cout << ans << '\n';
    return 0;
}

Compilation message (stderr)

cubeword.cpp: In function 'int main()':
cubeword.cpp:47:31: warning: array subscript has type 'char' [-Wchar-subscripts]
   47 |             cnt[s.size()][s[0]][ss[0]]++;
      |                               ^
cubeword.cpp:47:38: warning: array subscript has type 'char' [-Wchar-subscripts]
   47 |             cnt[s.size()][s[0]][ss[0]]++;
      |                                      ^
cubeword.cpp:51:31: warning: array subscript has type 'char' [-Wchar-subscripts]
   51 |             cnt[s.size()][s[0]][ss[0]]++;
      |                               ^
cubeword.cpp:51:38: warning: array subscript has type 'char' [-Wchar-subscripts]
   51 |             cnt[s.size()][s[0]][ss[0]]++;
      |                                      ^
cubeword.cpp:52:32: warning: array subscript has type 'char' [-Wchar-subscripts]
   52 |             cnt[s.size()][ss[0]][s[0]]++;
      |                                ^
cubeword.cpp:52:38: warning: array subscript has type 'char' [-Wchar-subscripts]
   52 |             cnt[s.size()][ss[0]][s[0]]++;
      |                                      ^

• Subtask 1: 21 / 21
• Subtask 2: 29 / 29
• Subtask 3: 34 / 34
• Subtask 4: 0 / 16