結果

問題 No.430 文字列検索
コンテスト
ユーザー k
提出日時 2021-03-21 01:44:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,375 bytes
コンパイル時間 1,869 ms
コンパイル使用メモリ 197,892 KB
最終ジャッジ日時 2025-01-19 20:47:51
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 2 WA * 2 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

class rolling_hash {
  int base;
  int mod;
  vector<unsigned long long> hash;
  vector<unsigned long long> pow;
  
public:
  rolling_hash(const string &s, int b = 1123, int m = 1e9+7) {
    base = b;
    mod = m;
    int n = s.length();
    hash.resize(n + 1);
    pow.resize(n + 1);
    pow[0] = 1;
    for (int i = 0; i < n; i++) {
      hash[i+1] = (base * hash[i] + s[i]) % mod;
      pow[i+1] = base * pow[i] % mod;
    }
  }

  // l, r are 0-based index
  unsigned long long get(int l, int r) {
    ++r;
    unsigned long long ret = hash[r] - hash[l] * pow[r - l] % mod;
    ret = (ret % mod + mod) % mod;
    return ret;
  }

  unsigned long long calc(const string &s) {
    unsigned long long ret = 0;
    for (char c: s)
      ret = (base * ret + c) % mod;
    return ret;
  }
};

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);

  string s;
  cin >> s;

  rolling_hash h(s);
  int m;
  cin >> m;
  int ret = 0;
  for (int i = 0; i < m; i++) {
    string t;
    cin >> t;
    unsigned long long g = h.calc(t);
    for (int j = 0; j + t.length() <= s.length(); j++) {
      if (g == h.get(j, j + t.length() - 1)) {
        bool ck = true;
        for (int k = 0; k < t.length(); k++)
          ck &= s[j+k] == t[k];
        if (ck)
          ++ret;
      }
    }
  }
  cout << ret << endl;
  
  return 0;
}
0