結果

問題 No.430 文字列検索
ユーザー Arutu_R2
提出日時 2020-08-25 21:30:16
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,127 bytes
コンパイル時間 193 ms
コンパイル使用メモリ 82,076 KB
実行使用メモリ 121,404 KB
最終ジャッジ日時 2024-11-10 00:46:18
合計ジャッジ時間 3,635 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 1 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

S = input()
N = len(S)
M = int(input())

base = 1007
mod = 2**61-1

hash = [0]*(N+1)
power = [1]*(N+1)
for i in range(1,N+1):
    hash[i] = (hash[i-1]*base + ord(S[i-1])) %mod
    power[i] = power[i-1]*base %mod

def RollingHash(l,r):
    return hash[r] - hash[l]*power[r-l]%mod


base2 = 2009
mod2 = 10**9+9

hash2 = [0]*(N+1)
power2 = [1]*(N+1)
for i in range(1,N+1):
    hash2[i] = (hash2[i-1]*base2 + ord(S[i-1])) %mod2
    power2[i] = power2[i-1]*base2 %mod2

def RollingHash2(l,r):
    return hash2[r] - hash2[l]*power2[r-l]%mod2

possible = [defaultdict(int) for _ in range(11)]
for i in range(10):
    for j in range(N+1-i):
        possible[i][(RollingHash(j,j+i),RollingHash2(j,j+i))] += 1



ans = 0
for i in range(M):
    C = input()
    hashC = 0
    hashC2 = 0
    for i in range(len(C)):
        hashC += ord(C[i])*base**(len(C)-1-i) %mod
        hashC %= mod
        hashC2 += ord(C[i])*base2**(len(C)-1-i) %mod2
        hashC2 %= mod2
    #if possible[len(C)][hashC] == possible2[len(C)][hashC]:
    ans += possible[len(C)][(hashC,hashC2)]
    #print(hashC)
    
print(ans)
0