62302023-11-08 12:06:03Error42Forgó rulettkerékcpp17Accepted 100/100890ms98624 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<ll(1e14) + 39>;

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

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

    mod fp = 1;
    for (int i = 0; i < m - 1; i++)
        fp *= 31;

    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
1Accepted3ms1832 KiB
2Accepted3ms2188 KiB
subtask220/20
3Accepted3ms2428 KiB
4Accepted3ms2428 KiB
5Accepted3ms2628 KiB
6Accepted6ms2860 KiB
7Accepted7ms4060 KiB
8Accepted6ms3176 KiB
9Accepted6ms3348 KiB
subtask315/15
10Accepted98ms5120 KiB
11Accepted119ms7332 KiB
12Accepted149ms12792 KiB
13Accepted101ms7208 KiB
14Accepted108ms8024 KiB
15Accepted250ms37512 KiB
16Accepted305ms48804 KiB
17Accepted268ms42692 KiB
subtask465/65
18Accepted356ms15460 KiB
19Accepted351ms10064 KiB
20Accepted695ms59984 KiB
21Accepted732ms53968 KiB
22Accepted479ms31748 KiB
23Accepted365ms10680 KiB
24Accepted633ms47204 KiB
25Accepted523ms25800 KiB
26Accepted216ms5464 KiB
27Accepted527ms32404 KiB
28Accepted890ms98624 KiB
29Accepted168ms11276 KiB
30Accepted523ms47596 KiB