結果

問題 No.430 文字列検索
コンテスト
ユーザー cormoran
提出日時 2016-10-18 17:25:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,669 bytes
コンパイル時間 1,462 ms
コンパイル使用メモリ 177,336 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2024-11-10 00:10:47
合計ジャッジ時間 4,802 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 4
other AC * 1 TLE * 1 -- * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using ll = long long;
#define rep(i, j) for(int i=0; i < (int)(j); i++)
#define repeat(i, j, k) for(int i = (j); i < (int)(k); i++)

// vector
template<class T> istream& operator >> (istream &is , vector<T> &v) { for(T &a : v) is >> a; return is; }

typedef unsigned long long ull;
const ull HashBase = 100000007;
vector<ull> pow_base;

vector<ull> build_rolling_hash(const string &s){
    int n = s.size();
    vector<ull> hash(n+1);
    pow_base.resize(n+1);
    hash[0] = 0;
    pow_base[0] = 1;
    rep(i, n){
        hash[i+1] = hash[i] * HashBase + s[i];
        pow_base[i+1] = pow_base[i] * HashBase;
    }
    return hash;
}

ull build_hash(const string &s) {
    ull base = 1;
    ull ret = 0;
    for(int i = s.size() - 1; i >= 0; i--) {
        ret += s[i] * base;
        base *= HashBase;
    }
    return ret;
}

// hash of (l,r] ( 1-based index)
ull get_rolling_hash(const vector<ull> &hash,int l,int r){
    return hash[r] - hash[l] * pow_base[r-l];
}

class Solver {
  public:
    bool solve() {
        string s; cin >> s;
        int m; cin >> m;
        unordered_set<ull> hash;
        rep(i, m) {
            string c; cin >> c;
            hash.insert(build_hash(c));
        }

        vector<ull> rh = build_rolling_hash(s);        
        ll ans = 0;
        repeat(i, 1, s.size() + 1) {
            repeat(j, 1, i + 1) {
                ans += hash.count(get_rolling_hash(rh, j - 1, i));
            }
        }
        
        cout << ans << endl;
        
        return 0;
    }
};

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    Solver s;
    s.solve();
    return 0;
}
0