結果

問題 No.430 文字列検索
ユーザー Rei TsukadaRei Tsukada
提出日時 2024-08-24 10:38:02
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,239 bytes
コンパイル時間 146 ms
コンパイル使用メモリ 82,216 KB
実行使用メモリ 82,988 KB
最終ジャッジ日時 2024-11-10 01:12:26
合計ジャッジ時間 3,565 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 32 ms
57,344 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 #

def rolling_hash(text: str, pattern: str) -> int:
    a = 31
    h = 998244353
    
    counter = 0
    
    text_length, pattern_length = len(text), len(pattern)
    text_hash = pattern_hash = 0
    
    # テキストよりもpatternの方が長い
    if text_length < pattern_length:
        return 0
    
    # a^l を先に計算
    
    a_l = 1
    for _ in range(pattern_length):
        a_l = (a_l * a) % h
        
    # 最初のハッシュを計算する
    for cursor in range(pattern_length):
        text_hash = (a * text_hash + ord(text[cursor])) % h
        pattern_hash = (a * pattern_hash + ord(pattern[cursor])) % h
    
    for i in range(text_length - pattern_length + 1):
        if text_hash == pattern_hash:
            counter += 1
        
        # 更新
        if i < text_length - pattern_length:
            text_hash = (text_hash * a - a_l * ord(text[i]) + ord(text[i + pattern_length])) % h
            
        if text_hash < 0:
            text_hash += h
            
    return counter

def main():
    S = input()
    M = int(input())
    
    ans = 0
    
    for _ in range(M):
        C = input()
        ans += rolling_hash(S, C)
    
    print(ans)
    
if __name__ == "__main__":
    main()
0