結果
問題 |
No.430 文字列検索
|
ユーザー |
|
提出日時 | 2022-02-06 16:53:18 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 904 bytes |
コンパイル時間 | 161 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 97,488 KB |
最終ジャッジ日時 | 2024-11-10 00:59:29 |
合計ジャッジ時間 | 2,832 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 10 WA * 4 |
ソースコード
# 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 = 4; mod = 1<<61-1 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)))