結果

問題 No.430 文字列検索
コンテスト
ユーザー Dente
提出日時 2020-02-21 19:41:14
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 2,566 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,627 ms
コンパイル使用メモリ 184,212 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-10 00:41:05
合計ジャッジ時間 2,576 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 1 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

//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;

/*
    基数として乱数をとりたい場合 : 
    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> rh(s, B_1, B_2, MOD_1, MOD_2);
*/
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, T B_1 = 1007, T B_2 = 1009, T MOD_1 = 1000000007, T MOD_2 = 1000000009)
     : B_1(B_1), B_2(B_2), MOD_1(MOD_1), MOD_2(MOD_2) { init(v); }

    RollingHash(string & s, T B_1 = 1007, T B_2 = 1009, T MOD_1 = 1000000007, T 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 P(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