srupのメモ帳

競プロで解いた問題や勉強したことを記録していくメモ帳

yukicoder No.430 文字列検索

問題

問題概要

文字列sの中に文字列cがいくつ含まれているかを求める問題。

解法

ローリングハッシュを使いました。ほぼ理解していないので、応用はきかないので要練習。文字列を比較するとき、1文字ずつ見比べいると、文字列の長さがmだとそこで、O(m)なってしまうが、ローリングハッシュを使用することで、そこをO(1)でできる。
わかりやすい解説スライド

www.slideshare.net

ミス

蟻本写経タイム。

コード

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <cstdio>
#include <set>
using namespace std;
typedef long long ll;
#define rep(i,n) for(int i=0;i<(n);i++)
const int INF = 1e9;

string c[5010];

typedef unsigned long long ull;
const ull B = 100000007;
//aはbにいくつ
int contain(string a, string b){
    int ret = 0;
    int al = a.length(), bl = b.length();
    if(al > bl) return 0;

    ull t = 1;
    for (int i = 0; i < al; ++i) t *= B;

    ull ah = 0, bh = 0;
    for (int i = 0; i < al; ++i) ah = ah * B + a[i];
    for (int i = 0; i < al; ++i) bh = bh * B + b[i];

    for (int i = 0; i + al <= bl; ++i){
        if(ah == bh) ret++;
        if(i + al < bl) bh = bh * B + b[i + al] - b[i] * t;
    }
    return ret;
}

int main(void){
    string s; cin >> s;
    int m; cin >> m;
    rep(i, m) cin >> c[i];
    ll ans = 0;
    rep(i, m){
        ans += contain(c[i], s);
    }
    printf("%lld\n", ans);
    return 0;
}