結果

問題 No.599 回文かい
コンテスト
ユーザー maspy
提出日時 2020-03-19 12:39:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 717 ms / 4,000 ms
コード長 893 bytes
コンパイル時間 408 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 80,512 KB
最終ジャッジ日時 2024-12-14 03:07:23
合計ジャッジ時間 4,590 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/ python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
MOD = 10 ** 9 + 7

S = read().rstrip()
N = len(S)


def z_algorithm(S):
    """compute the lengths of common prefix. """
    N = len(S)
    arr = [0] * N
    arr[0] = N
    i, j = 1, 0
    while i < N:
        while i + j < N and S[j] == S[i + j]:
            j += 1
        arr[i] = j
        if not j:
            i += 1
            continue
        k = 1
        while i + k < N and k + arr[k] < j:
            arr[i + k] = arr[k]
            k += 1
        i += k
        j -= k
    return arr


M = N // 2
dp = [1] * (M + 1)
for n in range(M, -1, -1):
    T = S[n: N - n]
    if not T:
        continue
    Z = z_algorithm(T)
    for i in range(1, len(T) // 2 + 1):
        if Z[-i] == i:
            dp[n] += dp[n + i]
    dp[n] %= MOD
print(dp[0])
0