結果

問題 No.430 文字列検索
コンテスト
ユーザー sekiya9311
提出日時 2016-10-05 01:26:54
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 204 ms / 2,000 ms
コード長 4,166 bytes
コンパイル時間 1,394 ms
コンパイル使用メモリ 129,644 KB
実行使用メモリ 42,880 KB
最終ジャッジ日時 2024-11-10 00:09:22
合計ジャッジ時間 3,205 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <list>
#include <vector>
#include <complex>
#include <utility>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <climits>
#include <bitset>
#include <ctime>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <cassert>
#include <cstddef>
#include <iomanip>
#include <numeric>
#include <tuple>
#include <sstream>
#include <fstream>
#include <random>

#define REP(i, n) for (int (i) = 0; (i) < (n); (i)++)
#define FOR(i, a, b) for (int (i) = (a); (i) < (b); (i)++)
#define RREP(i, a) for(int (i) = (a)-1; (i) >= 0; (i)--)
#define FORR(i, a, b) for(int (i) = (a)-1; (i) >= (b); (i)--)
#define PI acos(-1.0)
#define DEBUG(C) cerr << C << endl;
#define VI vector <int>
#define VII vector <VI>
#define VL vector <LL>
#define VLL vector <VL>
#define VD vector <double>
#define VDD vector <VD>
#define PII pair <int, int>
#define PDD pair <double, double>
#define PLL pair <LL, LL>
#define VPII vector <PII>
#define ALL(a) (a).begin(), (a).end()
#define SORT(a) sort(ALL(a))
#define REVERSE(a) reverse(ALL(a))
#define MP make_pair
#define EB emplace_back
#define FORE(a, b) for (auto &&a:b)
#define FIND(s, n) (s.find(n) != s.end())

using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
const int INF = 1e9;
const int MOD = INF + 7;
const LL LLINF = 1e18;

class RollingHash {
private:
    const long long mods[2] = {(long long)(1e9 + 7), (long long)(1e9 + 9)};
    vector <long long> hs[2], pm[2];
    vector <long long> x; //基数
    int len;
    string S;
public:
    //文字列, 基数1, 基数2
    RollingHash(const string& str, long long num1, long long num2) {
        x.resize(2);
        x[0] = num1; x[1] = num2;
        S = str;
        len = str.length();
        for (int i = 0; i < 2; i++) {
            pm[i].resize(1000000);
            pm[i][0] = 1;
            for (int j = 0; j < (int)pm[i].size(); j++) {
                pm[i][j + 1] = (pm[i][j] * x[i]) % mods[i];
            }
        }
        init(S);
    }
    void init(const string& str) {
        len = str.length();
        for (int i = 0; i < 2; i++) {
            hs[i].resize(len + 5);
            hs[i][0] = 0;
            for (int j  = 0; j < len; j++) {
                hs[i][j + 1] = ((hs[i][j] * x[i]) + str[j]) % mods[i];
            }
        }
    }
    //strのハッシュ値生成
    pair <long long, long long> makeHash(const string& str) {
        long long res[2];
        int slen = str.length();
        for (int i = 0; i < 2; i++) {
            res[i] = 0;
            for (int j = 0; j < slen; j++) {
                res[i] = ((res[i] * x[i]) + str[j]) % mods[i];
            }
        }
        return make_pair(res[0], res[1]);
    }
    //S[l..r]のハッシュ値返す
    pair <long long, long long> getHash(int l, int r) {
        if (l > r) return make_pair(-1, -1);
        long long res[2];
        while (pm[0].size() < r + 5) {
            for (int i = 0; i < 2; i++) {
                pm[i].emplace_back((pm[i].back() * x[i]) % mods[i]);
            }
        }
        for (int i = 0; i < 2; i++) {
            res[i] = (hs[i][r + 1] - (hs[i][l] * pm[i][r - l + 1] % mods[i]) + mods[i]) % mods[i];
        }
        return make_pair(res[0], res[1]);
    }
    //Sのハッシュ配列返す
    vector <vector <long long>> getHashArray() {
        return vector <vector <long long>>{hs[0], hs[1]};
    }
    vector <long long> getMod() {
        return vector <long long>{mods[0], mods[1]};
    }
    vector <long long> getBaseNum() {
        return x;
    }
};

int main(void) {
    string S; cin >> S;
    random_device rnd;
    RollingHash r(S, rnd() % MOD, rnd() % MOD);
    map<PLL, int> mp[10];
    REP(i, 10) {
        if (S.length() <= i) continue;
        REP(j, (int)S.length() - i)  mp[i][r.getHash(j, j + i)]++;
    }
    int M; cin >> M;
    int ans = 0;
    REP(i, M) {
        string s; cin >> s;
        if (s.length() > S.length()) continue;
        ans += mp[(int)s.length() - 1][r.makeHash(s)];
    }
    cout << ans << endl;
    return 0;
}
0