結果

問題 No.430 文字列検索
ユーザー kk
提出日時 2021-03-21 01:44:23
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,375 bytes
コンパイル時間 1,750 ms
コンパイル使用メモリ 204,136 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2024-11-10 00:54:05
合計ジャッジ時間 5,074 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
10,496 KB
testcase_01 TLE -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
権限があれば一括ダウンロードができます

ソースコード

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