結果

問題 No.430 文字列検索
ユーザー 里旬
提出日時 2019-03-14 04:33:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 331 ms / 2,000 ms
コード長 2,646 bytes
コンパイル時間 2,568 ms
コンパイル使用メモリ 205,360 KB
最終ジャッジ日時 2025-01-06 22:08:58
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

class rolling_hash{
    mt19937 mt;
    const size_t p_cnt = 3;
    const uint32_t p[3] = {2111511013, 2131131137, 2147483647};
    uint64_t b[3];

    uint64_t mpow(uint64_t x, uint64_t y, uint32_t m){
        if(y == 0) return 1;
        if(y == 1) return x%m;
        if(y%2 == 0) return mpow(x*x%m, y/2, m);
        return mpow(x*x%m, y/2, m) * x % m;
    }
    bool is_equal(const vector<uint32_t> &x, const vector<uint32_t> &y){
        for(size_t i=0;i<p_cnt;i++){
            if(x[i] != y[i]) return false;
        }
        return true;
    }
public:
    rolling_hash(){
        random_device rd;
        mt.seed(rd());
        for(size_t i=0;i<p_cnt;i++){
            b[i] = 0; 
            while(!b[i]) b[i] = mt() % p[i];
        }
    }
    vector<uint32_t> operator()(string s){
        vector<uint32_t> h = {0, 0, 0};
        for(char c : s){
            for(size_t i=0;i<p_cnt;i++) h[i] = (h[i]*b[i]%p[i]+c) % p[i];
        }
        return h;
    }
    // x が y の中に何回出現するか
    uint32_t contain(const vector<string> &xs, const string &y){
        if(xs.size() == 0) return 0;
        size_t xl = xs[0].size(), yl = y.size();
        if(xl > yl) return 0;
        uint64_t bl[p_cnt];
        for(size_t i=0;i<p_cnt;i++) bl[i] = mpow(b[i], xl, p[i]);

        vector<vector<uint32_t>> xhs(xs.size());
        for(size_t i=0;i<xs.size();i++) xhs[i] = vector<uint32_t>(p_cnt, 0);
        vector<uint32_t> yh(3, 0);
        for(size_t i=0;i<p_cnt;i++){
            for(size_t xi=0;xi<xs.size();xi++)
                for(size_t j=0;j<xl;j++) xhs[xi][i] = (xhs[xi][i]*b[i]%p[i] + xs[xi][j]) % p[i];
            for(size_t j=0;j<xl;j++) yh[i] = (yh[i]*b[i]%p[i] + y[j]) % p[i];
        }

        uint32_t cnt = 0;
        for(size_t pos=0;pos+xl<=yl;pos++){
            for(size_t xi=0;xi<xs.size();xi++){
                if(is_equal(xhs[xi], yh)) cnt++;
            }
            if(pos + xl < yl)
                for(size_t i=0;i<p_cnt;i++) yh[i] = (yh[i]*b[i]%p[i] + y[xl+pos] +p[i] - y[pos]*bl[i]%p[i]) % p[i];
        }

        return cnt;
    }
};

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    cout.precision(12);
    cout.setf(ios_base::fixed, ios_base::floatfield);
    
    rolling_hash rh;
    string s, c[5000];
    int m;

    cin >> s;
    cin >> m;
    for(int i=0;i<m;i++) cin >> c[i];

    vector<string> cs[11];
    for(int i=0;i<m;i++){
        cs[c[i].size()].push_back(c[i]);
    }

    uint64_t ans = 0;
    for(int i=1;i<=10;i++){
        ans += rh.contain(cs[i], s);
    }
    cout << ans << endl;
    
    return 0;
}
0