62292023-11-08 12:04:51Error42Forgó rulettkerékcpp17Wrong answer 35/100906ms98832 KiB
#include <iostream>
#include <map>
#include <set>
#include <vector>

using namespace std;

using ll = long long;

template <ll M>
struct modular {
    ll val;

    // n >= 0
    modular(const ll n) {
        if (n < M)
            val = n;
        else if (n < 2 * M)
            val = n - M;
        else
            val = n % M;
    }

    [[nodiscard]] modular operator +(const modular& rhs) const {
        return val + rhs.val;
    }

    [[nodiscard]] modular operator -(const modular& rhs) const {
        return val + M - rhs.val;
    }

    [[nodiscard]] modular operator *(const modular& rhs) const {
        return val * rhs.val;
    }

    // p >= 0
    [[nodiscard]] modular pow(const ll& p) const {
        if (p == 0)
            return 1;

        if (p % 2 == 0)
            return (*this * *this).pow(p / 2);
        else
            return *this * pow(p - 1);
    }

    [[nodiscard]] modular inv() const {
        return pow(M - 2);
    }

    [[nodiscard]] modular operator /(const modular& rhs) const {
        return *this * rhs.inv();
    }

    modular& operator +=(const modular& rhs) {
        return *this = *this + rhs;
    }

    modular& operator -=(const modular& rhs) {
        return *this = *this - rhs;
    }

    modular& operator *=(const modular& rhs) {
        return *this = *this * rhs;
    }

    modular& operator /=(const modular& rhs) {
        return *this = *this / rhs;
    }

    explicit operator ll() const {
        return val;
    }
};

template <ll M>
ostream& operator<<(ostream& os, const modular<M>& modular) {
    cout << modular.val;
    return os;
}

using mod = modular<1'000'000'007>;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    ll n, m;
    cin >> n >> m;

    mod const fp = mod(31).pow(m - 1);

    map<ll, ll> hashes;
    ll ans = 0;

    for (int w = 0; w < n; w++) {
        string s;
        cin >> s;

        mod cur_hash = 0;

        for (char const& c : s) {
            cur_hash *= 31;
            cur_hash += c - 'a' + 1;
        }

        ans += hashes[ll(cur_hash)];

        set<ll> cur_hashes;
        cur_hashes.insert(ll(cur_hash));

        for (int i = 0; i < m - 1; i++) {
            cur_hash -= fp * (s[i] - 'a' + 1);
            cur_hash *= 31;
            cur_hash += (s[i] - 'a' + 1);

            cur_hashes.insert(ll(cur_hash));
        }

        for (ll const h : cur_hashes)
            hashes[ll(h)]++;
    }

    cout << ans << "\n";
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Accepted3ms1700 KiB
2Accepted3ms1864 KiB
subtask220/20
3Accepted3ms2104 KiB
4Accepted3ms2312 KiB
5Accepted4ms2696 KiB
6Accepted6ms2948 KiB
7Accepted7ms3888 KiB
8Accepted6ms3176 KiB
9Accepted6ms3312 KiB
subtask315/15
10Accepted94ms5140 KiB
11Accepted118ms7300 KiB
12Accepted143ms12516 KiB
13Accepted98ms7104 KiB
14Accepted107ms8036 KiB
15Accepted250ms37688 KiB
16Accepted300ms48976 KiB
17Accepted246ms42568 KiB
subtask40/65
18Accepted356ms15340 KiB
19Accepted344ms10104 KiB
20Accepted690ms60204 KiB
21Wrong answer720ms54144 KiB
22Accepted455ms31888 KiB
23Accepted370ms11292 KiB
24Accepted632ms47648 KiB
25Accepted501ms26188 KiB
26Accepted218ms5508 KiB
27Wrong answer536ms32308 KiB
28Accepted906ms98832 KiB
29Accepted163ms11644 KiB
30Accepted514ms47928 KiB