結果

問題 No.430 文字列検索
ユーザー DenteDente
提出日時 2020-07-05 17:28:01
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 184 ms / 2,000 ms
コード長 2,397 bytes
コンパイル時間 2,223 ms
コンパイル使用メモリ 217,252 KB
実行使用メモリ 28,032 KB
最終ジャッジ日時 2024-11-10 00:44:50
合計ジャッジ時間 3,661 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 184 ms
28,032 KB
testcase_02 AC 33 ms
5,504 KB
testcase_03 AC 34 ms
5,504 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 172 ms
27,648 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 14 ms
5,760 KB
testcase_11 AC 71 ms
9,088 KB
testcase_12 AC 72 ms
8,960 KB
testcase_13 AC 72 ms
8,960 KB
testcase_14 AC 59 ms
7,424 KB
testcase_15 AC 52 ms
6,656 KB
testcase_16 AC 36 ms
5,376 KB
testcase_17 AC 34 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// yukicoder-430
#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (n); i++)
#define ALL(v) (v).begin(), (v).end()
using ll = long long;
using P = pair<int, int>;
const int INF = 1e9;
const long long LINF = 1e18;
const long long MOD = 1e9 + 7;

template <typename T>
struct RollingHash {
    using ll = long long;
    ll B_1, B_2, MOD_1, MOD_2;
    vector<ll> hash_1, hash_2, pow_1, pow_2;

    RollingHash(vector<T> v, ll B_1 = 1007, ll B_2 = 1009,
                ll MOD_1 = 1000000007, ll MOD_2 = 1000000009)
        : B_1(B_1), B_2(B_2), MOD_1(MOD_1), MOD_2(MOD_2) {
        init(v);
    }

    RollingHash(string& s, ll B_1 = 1007, ll B_2 = 1009, ll MOD_1 = 1000000007,
                ll MOD_2 = 1000000009)
        : B_1(B_1), B_2(B_2), MOD_1(MOD_1), MOD_2(MOD_2) {
        vector<T> v;
        for (char c : s) v.emplace_back(c);
        init(v);
    }

    void init(vector<T> v) {
        int n = v.size();
        hash_1.assign(n + 1, 0);
        hash_2.assign(n + 1, 0);
        pow_1.assign(n + 1, 1);
        pow_2.assign(n + 1, 1);
        for (int i = 0; i < n; i++) {
            hash_1[i + 1] = (hash_1[i] * B_1 + v[i]) % MOD_1;
            hash_2[i + 1] = (hash_2[i] * B_2 + v[i]) % MOD_2;
            pow_1[i + 1] = pow_1[i] * B_1 % MOD_1;
            pow_2[i + 1] = pow_2[i] * B_2 % MOD_2;
        }
    }

    pair<ll, ll> get(int l, int r) {
        ll res_1 =
            ((hash_1[r] - hash_1[l] * pow_1[r - l]) % MOD_1 + MOD_1) % MOD_1;
        ll res_2 =
            ((hash_2[r] - hash_2[l] * pow_2[r - l]) % MOD_2 + MOD_2) % MOD_2;
        return pair<ll, ll>(res_1, res_2);
    }
};

signed main() {
    string s;
    cin >> s;
    int n = s.size();
    int m;
    cin >> m;
    string c[m];
    rep(i, m) {
        cin >> c[i];
    }
    ll B_1, B_2, MOD_1 = 1000000007, MOD_2 = 1000000009;
    random_device rnd;
    mt19937 mt(rnd());
    B_1 = mt() % (MOD_1 - 2) + 2;
    B_2 = mt() % (MOD_2 - 2) + 2;
    RollingHash<char> rhs(s, B_1, B_2, MOD_1, MOD_2);
    vector<map<pair<ll, ll>, int>> mp(11);
    for (int i = 1; i <= 10; i++) {
        for (int j = 0; j + i <= n; j++) {
            mp[i][rhs.get(j, j + i)]++;
        }
    }
    int ans = 0;
    rep(i, m) {
        RollingHash<char> rhc(c[i], B_1, B_2, MOD_1, MOD_2);
        ans += mp[c[i].size()][rhc.get(0, c[i].size())];
    }
    cout << ans << endl;
    return 0;
}
0