結果
問題 |
No.430 文字列検索
|
ユーザー |
|
提出日時 | 2022-02-06 16:54:11 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 205 ms / 2,000 ms |
コード長 | 905 bytes |
コンパイル時間 | 161 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 141,696 KB |
最終ジャッジ日時 | 2024-11-10 00:59:31 |
合計ジャッジ時間 | 2,571 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 14 |
ソースコード
# 1-dimension Rolling Hash # 参考 https://tjkendev.github.io/procon-library/python/string/rolling_hash.html class RollingHash(): def __init__(self, s, base, mod): self.mod = mod self.pw = pw = [1]*(len(s)+1) self.length = l = len(s) self.h = h = [0]*(l+1) v = 0 for i in range(l): h[i+1] = v = (v * base + ord(s[i])) % mod v = 1 for i in range(l): pw[i+1] = v = v * base % mod def get(self, l, r): return (self.h[r] - self.h[l] * self.pw[r-l]) % self.mod def all(self): return self.get(0, self.length) base = 37; mod = 10**9+7 S = input() N = len(S) M = int(input()) RHS = RollingHash(S,base,mod) from collections import defaultdict dic = defaultdict(lambda: 0) for l in range(1,11): for i in range(N-l+1):dic[RHS.get(i,i+l)]+=1 print(sum(dic[RollingHash(input(),base,mod).all()]for _ in range(M)))