結果

問題 No.430 文字列検索
ユーザー AEnAEn
提出日時 2022-12-10 17:08:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 593 ms / 2,000 ms
コード長 902 bytes
コンパイル時間 240 ms
コンパイル使用メモリ 82,120 KB
実行使用メモリ 183,368 KB
最終ジャッジ日時 2024-11-10 01:02:57
合計ジャッジ時間 5,833 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
53,888 KB
testcase_01 AC 593 ms
183,368 KB
testcase_02 AC 337 ms
78,924 KB
testcase_03 AC 334 ms
79,024 KB
testcase_04 AC 38 ms
54,272 KB
testcase_05 AC 38 ms
54,016 KB
testcase_06 AC 35 ms
54,272 KB
testcase_07 AC 38 ms
54,272 KB
testcase_08 AC 543 ms
182,188 KB
testcase_09 AC 44 ms
62,080 KB
testcase_10 AC 96 ms
86,996 KB
testcase_11 AC 449 ms
97,412 KB
testcase_12 AC 457 ms
97,536 KB
testcase_13 AC 457 ms
97,408 KB
testcase_14 AC 411 ms
90,520 KB
testcase_15 AC 384 ms
85,888 KB
testcase_16 AC 345 ms
79,512 KB
testcase_17 AC 355 ms
79,000 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict


S = input()
S = [ord(i)-96 for i in S]

l = len(S)
b1 = 10**9+7
b2 = 10**5+7
mod = (1<<61)-1
p1, p2, h1, h2 = [1]*(l+1),[1]*(l+1),[0]*(l+1),[0]*(l+1)
for i in range(l):
    p1[i+1] = (p1[i]*b1)%mod
    p2[i+1] = (p2[i]*b2)%mod
    h1[i+1] = (h1[i]*b1+S[i])%mod
    h2[i+1] = (h2[i]*b2+S[i])%mod

def gethash(l,r):
    """2つの基数でのハッシュ値を返す"""
    a = (h1[r]-h1[l]*p1[r-l])%mod
    b = (h2[r]-h2[l]*p2[r-l])%mod
    return a, b

d = defaultdict(int)
for i in range(l):
    for j in range(1,11):
        if i+j<=l:
            a, b = gethash(i,i+j)
            d[str(a)+'_'+str(b)] += 1
M = int(input())
res = 0
for i in range(M):
    C = input()
    C = [ord(i)-96 for i in C]
    hash1, hash2 = 0,0
    for num in C:
        hash1 = (hash1*b1+num)%mod
        hash2 = (hash2*b2+num)%mod
    res += d[str(hash1)+'_'+str(hash2)]
print(res)

0