oj.uz

Submission #572387

# Submission time Handle Problem Language Result Execution time Memory
572387 2022-06-04T10:27:55 Z piOOE Cubeword (CEOI19_cubeword) C++17
50 / 100
 141 ms 14252 KB 
#include <bits/stdc++.h>

using namespace std;

#define sz(x) ((int)size(x))
#define all(x) begin(x), end(x)
#define trace(x) cout << #x << ": " << (x) << endl;

namespace Ment {
    template<typename T>
    T inverse(T a, T m) {
        T u = 0, v = 1;
        while (a != 0) {
            T t = m / a;
            m -= t * a;
            swap(a, m);
            u -= t * v;
            swap(u, v);
        }
        assert(m == 1);
        return u;
    }

    template<typename T>
    class Modular {
    public:
        using Type = typename decay<decltype(T::value)>::type;

        constexpr Modular() : value() {}

        template<typename U>
        Modular(const U &x) {
            value = normalize(x);
        }

        template<typename U>
        static Type normalize(const U &x) {
            Type v;
            if (-mod() <= x && x < mod()) v = static_cast<Type>(x);
            else v = static_cast<Type>(x % mod());
            if (v < 0) v += mod();
            return v;
        }

        const Type &operator()() const { return value; }

        template<typename U>
        explicit operator U() const { return static_cast<U>(value); }

        constexpr static Type mod() { return T::value; }

        Modular &operator+=(const Modular &other) {
            value += other.value - mod();
            value += (value >> 31) & mod();
            return *this;
        }

        Modular &operator-=(const Modular &other) {
            value -= other.value;
            value += (value >> 31) & mod();
            return *this;
        }

        template<typename U>
        Modular &operator+=(const U &other) { return *this += Modular(other); }

        template<typename U>
        Modular &operator-=(const U &other) { return *this -= Modular(other); }

        Modular &operator++() { return *this += 1; }

        Modular &operator--() { return *this -= 1; }

        Modular operator++(int) {
            Modular result(*this);
            *this += 1;
            return result;
        }

        Modular operator--(int) {
            Modular result(*this);
            *this -= 1;
            return result;
        }

        Modular operator-() const { return Modular(-value); }

        template<typename U = T>
        typename enable_if<is_same<typename Modular<U>::Type, int>::value, Modular>::type &
        operator*=(const Modular &rhs) {
#ifdef _WIN32
            uint64_t x = static_cast<int64_t>(value) * static_cast<int64_t>(rhs.value);
            uint32_t xh = static_cast<uint32_t>(x >> 32), xl = static_cast<uint32_t>(x), d, m;
            asm(
                    "divl %4; \n\t"
                    : "=a" (d), "=d" (m)
                    : "d" (xh), "a" (xl), "r" (mod())
                    );
            value = m;
#else
            value = normalize(static_cast<int64_t>(value) * static_cast<int64_t>(rhs.value));
#endif
            return *this;
        }

        template<typename U = T>
        typename enable_if<is_same<typename Modular<U>::Type, int64_t>::value, Modular>::type &
        operator*=(const Modular &rhs) {
            int64_t q = static_cast<int64_t>(static_cast<long double>(value) * rhs.value / mod());
            value = normalize(value * rhs.value - q * mod());
            return *this;
        }

        template<typename U = T>
        typename enable_if<!is_integral<typename Modular<U>::Type>::value, Modular>::type &
        operator*=(const Modular &rhs) {
            value = normalize(value * rhs.value);
            return *this;
        }

        Modular &operator/=(const Modular &other) { return *this *= Modular(inverse(other.value, mod())); }

        template<typename U>
        friend bool operator==(const Modular<U> &lhs, const Modular<U> &rhs);

        template<typename U>
        friend bool operator<(const Modular<U> &lhs, const Modular<U> &rhs);

        template<typename U>
        friend std::istream &operator>>(std::istream &stream, Modular<U> &number);

    private:
        Type value;
    };

    template<typename T>
    bool operator==(const Modular<T> &lhs, const Modular<T> &rhs) { return lhs.value == rhs.value; }

    template<typename T, typename U>
    bool operator==(const Modular<T> &lhs, U rhs) { return lhs == Modular<T>(rhs); }

    template<typename T, typename U>
    bool operator==(U lhs, const Modular<T> &rhs) { return Modular<T>(lhs) == rhs; }

    template<typename T>
    bool operator!=(const Modular<T> &lhs, const Modular<T> &rhs) { return !(lhs == rhs); }

    template<typename T, typename U>
    bool operator!=(const Modular<T> &lhs, U rhs) { return !(lhs == rhs); }

    template<typename T, typename U>
    bool operator!=(U lhs, const Modular<T> &rhs) { return !(lhs == rhs); }

    template<typename T>
    bool operator<(const Modular<T> &lhs, const Modular<T> &rhs) { return lhs.value < rhs.value; }

    template<typename T>
    Modular<T> operator+(const Modular<T> &lhs, const Modular<T> &rhs) { return Modular<T>(lhs) += rhs; }

    template<typename T, typename U>
    Modular<T> operator+(const Modular<T> &lhs, U rhs) { return Modular<T>(lhs) += rhs; }

    template<typename T, typename U>
    Modular<T> operator+(U lhs, const Modular<T> &rhs) { return Modular<T>(lhs) += rhs; }

    template<typename T>
    Modular<T> operator-(const Modular<T> &lhs, const Modular<T> &rhs) { return Modular<T>(lhs) -= rhs; }

    template<typename T, typename U>
    Modular<T> operator-(const Modular<T> &lhs, U rhs) { return Modular<T>(lhs) -= rhs; }

    template<typename T, typename U>
    Modular<T> operator-(U lhs, const Modular<T> &rhs) { return Modular<T>(lhs) -= rhs; }

    template<typename T>
    Modular<T> operator*(const Modular<T> &lhs, const Modular<T> &rhs) { return Modular<T>(lhs) *= rhs; }

    template<typename T, typename U>
    Modular<T> operator*(const Modular<T> &lhs, U rhs) { return Modular<T>(lhs) *= rhs; }

    template<typename T, typename U>
    Modular<T> operator*(U lhs, const Modular<T> &rhs) { return Modular<T>(lhs) *= rhs; }

    template<typename T>
    Modular<T> operator/(const Modular<T> &lhs, const Modular<T> &rhs) { return Modular<T>(lhs) /= rhs; }

    template<typename T, typename U>
    Modular<T> operator/(const Modular<T> &lhs, U rhs) { return Modular<T>(lhs) /= rhs; }

    template<typename T, typename U>
    Modular<T> operator/(U lhs, const Modular<T> &rhs) { return Modular<T>(lhs) /= rhs; }

    template<typename T, typename U>
    Modular<T> power(const Modular<T> &a, const U &b) {
        assert(b >= 0);
        Modular<T> x = a, res = 1;
        U p = b;
        while (p > 0) {
            if (p & 1) res *= x;
            x *= x;
            p >>= 1;
        }
        return res;
    }

    template<typename T>
    string to_string(const Modular<T> &number) {
        return to_string(number());
    }

    template<typename T>
    std::ostream &operator<<(std::ostream &stream, const Modular<T> &number) {
        return stream << number();
    }

    template<typename T>
    std::istream &operator>>(std::istream &stream, Modular<T> &number) {
        typename common_type<typename Modular<T>::Type, int64_t>::type x;
        stream >> x;
        number.value = Modular<T>::normalize(x);
        return stream;
    }

    constexpr int md = 998244353;
    using Mint = Modular<std::integral_constant<decay<decltype(md)>::type, md>>;
}
using Ment::Mint;


typedef long long ll;

mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

int rand(int l, int r) { return (int) ((ll) rnd() % (r - l + 1)) + l; }

const int N = 8, A = 16, L = 11;
const ll infL = 3e18;

int g[N][3] = {{1, 2, 4},
               {0, 3, 5},
               {0, 3, 6},
               {1, 2, 7},
               {0, 5, 6},
               {1, 4, 7},
               {2, 4, 7},
               {3, 5, 6}};

int n;
Mint cnt[A][A][L], dp2[A][A][A], dp7[A][A][A], dp1[A][A][A], dp4[A][A][A];
Mint ans = 0;
int len = 0, used[N];

int get(char c) {
    if ('a' <= c && c <= 'z') {
        return c - 'a';
    }
    if ('A' <= c && c <= 'Z') {
        return (c - 'A') + 26;
    }
    return (c - '0') + 52;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cin >> n;
    vector<string> stt;
    for (int i = 0; i < n; ++i) {
        string s;
        cin >> s;
        string ss = s;
        reverse(all(ss));
        stt.push_back(ss);
        stt.push_back(s);
    }
    sort(all(stt));
    stt.resize(unique(all(stt)) - begin(stt));
    for (string s: stt) {
        ++cnt[get(s[0])][get(s.back())][sz(s)];
    }
    for (len = 1; len < L; ++len) {
        {
            //2
            memset(used, -1, sizeof(used));
            for (int x0 = 0; x0 < A; ++x0) {
                used[0] = x0;
                for (int x3 = 0; x3 < A; ++x3) {
                    used[3] = x3;
                    for (int x6 = 0; x6 < A; ++x6) {
                        used[6] = x6;
                        Mint sum = 0;
                        for (int x = 0; x < A; ++x) {
                            Mint now = 1;
                            for (int to: g[2]) {
                                now *= cnt[x][used[to]][len];
                            }
                            sum = (sum + now);
                        }
                        dp2[x0][x3][x6] = sum;
                    }
                }
            }
        }
        {
            //1
            memset(used, -1, sizeof(used));
            for (int x0 = 0; x0 < A; ++x0) {
                used[0] = x0;
                for (int x3 = 0; x3 < A; ++x3) {
                    used[3] = x3;
                    for (int x5 = 0; x5 < A; ++x5) {
                        used[5] = x5;
                        Mint sum = 0;
                        for (int x = 0; x < A; ++x) {
                            Mint now = 1;
                            for (int to: g[1]) {
                                now *= cnt[x][used[to]][len];
                            }
                            sum += now;
                        }
                        dp1[x0][x3][x5] = sum;
                    }
                }
            }
        }
        {
            //7
            memset(used, -1, sizeof(used));
            for (int x6 = 0; x6 < A; ++x6) {
                used[6] = x6;
                for (int x3 = 0; x3 < A; ++x3) {
                    used[3] = x3;
                    for (int x5 = 0; x5 < A; ++x5) {
                        used[5] = x5;
                        Mint sum = 0;
                        for (int x = 0; x < A; ++x) {
                            Mint now = 1;
                            for (int to: g[7]) {
                                now *= cnt[x][used[to]][len];
                            }
                            sum += now;
                        }
                        dp7[x3][x5][x6] = sum;
                    }
                }
            }
        }
        {
            //4
            memset(used, -1, sizeof(used));
            for (int x0 = 0; x0 < A; ++x0) {
                used[0] = x0;
                for (int x6 = 0; x6 < A; ++x6) {
                    used[6] = x6;
                    for (int x5 = 0; x5 < A; ++x5) {
                        used[5] = x5;
                        Mint sum = 0;
                        for (int x = 0; x < A; ++x) {
                            Mint now = 1;
                            for (int to: g[4]) {
                                now *= cnt[x][used[to]][len];
                            }
                            sum += now;
                        }
                        dp4[x0][x5][x6] = sum;
                    }
                }
            }
        }
        for (int x0 = 0; x0 < A; ++x0) {
            for (int x3 = 0; x3 < A; ++x3) {
                for (int x5 = 0; x5 < A; ++x5) {
                    for (int x6 = 0; x6 < A; ++x6) {
                        ans = (ans + dp4[x0][x5][x6] * dp7[x3][x5][x6] * dp1[x0][x3][x5] *
                                     dp2[x0][x3][x6]);
                    }
                }
            }
        }
    }
    cout << ans;
    return 0;
}

Subtask #1
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 100 ms 9160 KB Output is correct
2 Correct 98 ms 9344 KB Output is correct
3 Correct 96 ms 9440 KB Output is correct
4 Correct 95 ms 9416 KB Output is correct
5 Correct 95 ms 9416 KB Output is correct
6 Correct 95 ms 9384 KB Output is correct
7 Correct 141 ms 9352 KB Output is correct
8 Correct 100 ms 9340 KB Output is correct
9 Correct 94 ms 9392 KB Output is correct
10 Correct 98 ms 9416 KB Output is correct
11 Correct 97 ms 9360 KB Output is correct
12 Correct 97 ms 9428 KB Output is correct
13 Correct 98 ms 9416 KB Output is correct
14 Correct 96 ms 9412 KB Output is correct
15 Correct 97 ms 9448 KB Output is correct
16 Correct 97 ms 9416 KB Output is correct
17 Correct 104 ms 9412 KB Output is correct
18 Correct 92 ms 9376 KB Output is correct
19 Correct 94 ms 9416 KB Output is correct
20 Correct 97 ms 9520 KB Output is correct
21 Correct 104 ms 9428 KB Output is correct

Subtask #2
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 100 ms 9160 KB Output is correct
2 Correct 98 ms 9344 KB Output is correct
3 Correct 96 ms 9440 KB Output is correct
4 Correct 95 ms 9416 KB Output is correct
5 Correct 95 ms 9416 KB Output is correct
6 Correct 95 ms 9384 KB Output is correct
7 Correct 141 ms 9352 KB Output is correct
8 Correct 100 ms 9340 KB Output is correct
9 Correct 94 ms 9392 KB Output is correct
10 Correct 98 ms 9416 KB Output is correct
11 Correct 97 ms 9360 KB Output is correct
12 Correct 97 ms 9428 KB Output is correct
13 Correct 98 ms 9416 KB Output is correct
14 Correct 96 ms 9412 KB Output is correct
15 Correct 97 ms 9448 KB Output is correct
16 Correct 97 ms 9416 KB Output is correct
17 Correct 104 ms 9412 KB Output is correct
18 Correct 92 ms 9376 KB Output is correct
19 Correct 94 ms 9416 KB Output is correct
20 Correct 97 ms 9520 KB Output is correct
21 Correct 104 ms 9428 KB Output is correct
22 Correct 109 ms 9428 KB Output is correct
23 Correct 93 ms 8692 KB Output is correct
24 Correct 99 ms 8720 KB Output is correct
25 Correct 112 ms 8776 KB Output is correct
26 Correct 97 ms 8840 KB Output is correct
27 Correct 93 ms 8804 KB Output is correct
28 Correct 98 ms 8740 KB Output is correct
29 Correct 104 ms 8792 KB Output is correct
30 Correct 114 ms 9336 KB Output is correct
31 Correct 94 ms 9420 KB Output is correct
32 Correct 93 ms 9416 KB Output is correct
33 Correct 102 ms 9416 KB Output is correct
34 Correct 105 ms 9408 KB Output is correct
35 Correct 95 ms 9328 KB Output is correct
36 Correct 91 ms 9412 KB Output is correct
37 Correct 103 ms 9424 KB Output is correct
38 Correct 100 ms 9416 KB Output is correct
39 Correct 100 ms 9424 KB Output is correct
40 Correct 95 ms 9416 KB Output is correct
41 Correct 102 ms 9360 KB Output is correct
42 Correct 109 ms 9416 KB Output is correct

Subtask #3
0 / 34.0

# Verdict Execution time Memory Grader output
1 Correct 100 ms 9160 KB Output is correct
2 Correct 98 ms 9344 KB Output is correct
3 Correct 96 ms 9440 KB Output is correct
4 Correct 95 ms 9416 KB Output is correct
5 Correct 95 ms 9416 KB Output is correct
6 Correct 95 ms 9384 KB Output is correct
7 Correct 141 ms 9352 KB Output is correct
8 Correct 100 ms 9340 KB Output is correct
9 Correct 94 ms 9392 KB Output is correct
10 Correct 98 ms 9416 KB Output is correct
11 Correct 97 ms 9360 KB Output is correct
12 Correct 97 ms 9428 KB Output is correct
13 Correct 98 ms 9416 KB Output is correct
14 Correct 96 ms 9412 KB Output is correct
15 Correct 97 ms 9448 KB Output is correct
16 Correct 97 ms 9416 KB Output is correct
17 Correct 104 ms 9412 KB Output is correct
18 Correct 92 ms 9376 KB Output is correct
19 Correct 94 ms 9416 KB Output is correct
20 Correct 97 ms 9520 KB Output is correct
21 Correct 104 ms 9428 KB Output is correct
22 Correct 109 ms 9428 KB Output is correct
23 Correct 93 ms 8692 KB Output is correct
24 Correct 99 ms 8720 KB Output is correct
25 Correct 112 ms 8776 KB Output is correct
26 Correct 97 ms 8840 KB Output is correct
27 Correct 93 ms 8804 KB Output is correct
28 Correct 98 ms 8740 KB Output is correct
29 Correct 104 ms 8792 KB Output is correct
30 Correct 114 ms 9336 KB Output is correct
31 Correct 94 ms 9420 KB Output is correct
32 Correct 93 ms 9416 KB Output is correct
33 Correct 102 ms 9416 KB Output is correct
34 Correct 105 ms 9408 KB Output is correct
35 Correct 95 ms 9328 KB Output is correct
36 Correct 91 ms 9412 KB Output is correct
37 Correct 103 ms 9424 KB Output is correct
38 Correct 100 ms 9416 KB Output is correct
39 Correct 100 ms 9424 KB Output is correct
40 Correct 95 ms 9416 KB Output is correct
41 Correct 102 ms 9360 KB Output is correct
42 Correct 109 ms 9416 KB Output is correct
43 Runtime error 102 ms 14252 KB Execution killed with signal 11
44 Halted 0 ms 0 KB -

Subtask #4
0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 100 ms 9160 KB Output is correct
2 Correct 98 ms 9344 KB Output is correct
3 Correct 96 ms 9440 KB Output is correct
4 Correct 95 ms 9416 KB Output is correct
5 Correct 95 ms 9416 KB Output is correct
6 Correct 95 ms 9384 KB Output is correct
7 Correct 141 ms 9352 KB Output is correct
8 Correct 100 ms 9340 KB Output is correct
9 Correct 94 ms 9392 KB Output is correct
10 Correct 98 ms 9416 KB Output is correct
11 Correct 97 ms 9360 KB Output is correct
12 Correct 97 ms 9428 KB Output is correct
13 Correct 98 ms 9416 KB Output is correct
14 Correct 96 ms 9412 KB Output is correct
15 Correct 97 ms 9448 KB Output is correct
16 Correct 97 ms 9416 KB Output is correct
17 Correct 104 ms 9412 KB Output is correct
18 Correct 92 ms 9376 KB Output is correct
19 Correct 94 ms 9416 KB Output is correct
20 Correct 97 ms 9520 KB Output is correct
21 Correct 104 ms 9428 KB Output is correct
22 Correct 109 ms 9428 KB Output is correct
23 Correct 93 ms 8692 KB Output is correct
24 Correct 99 ms 8720 KB Output is correct
25 Correct 112 ms 8776 KB Output is correct
26 Correct 97 ms 8840 KB Output is correct
27 Correct 93 ms 8804 KB Output is correct
28 Correct 98 ms 8740 KB Output is correct
29 Correct 104 ms 8792 KB Output is correct
30 Correct 114 ms 9336 KB Output is correct
31 Correct 94 ms 9420 KB Output is correct
32 Correct 93 ms 9416 KB Output is correct
33 Correct 102 ms 9416 KB Output is correct
34 Correct 105 ms 9408 KB Output is correct
35 Correct 95 ms 9328 KB Output is correct
36 Correct 91 ms 9412 KB Output is correct
37 Correct 103 ms 9424 KB Output is correct
38 Correct 100 ms 9416 KB Output is correct
39 Correct 100 ms 9424 KB Output is correct
40 Correct 95 ms 9416 KB Output is correct
41 Correct 102 ms 9360 KB Output is correct
42 Correct 109 ms 9416 KB Output is correct
43 Runtime error 102 ms 14252 KB Execution killed with signal 11
44 Halted 0 ms 0 KB -