結果

問題 No.599 回文かい
ユーザー maspymaspy
提出日時 2020-03-19 12:39:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 771 ms / 4,000 ms
コード長 893 bytes
コンパイル時間 832 ms
コンパイル使用メモリ 86,964 KB
実行使用メモリ 82,028 KB
最終ジャッジ日時 2023-08-20 20:46:22
合計ジャッジ時間 6,582 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 75 ms
71,308 KB
testcase_01 AC 74 ms
71,044 KB
testcase_02 AC 75 ms
71,404 KB
testcase_03 AC 75 ms
71,032 KB
testcase_04 AC 105 ms
76,032 KB
testcase_05 AC 86 ms
75,980 KB
testcase_06 AC 93 ms
76,028 KB
testcase_07 AC 90 ms
75,996 KB
testcase_08 AC 89 ms
76,048 KB
testcase_09 AC 87 ms
76,068 KB
testcase_10 AC 209 ms
77,496 KB
testcase_11 AC 176 ms
77,476 KB
testcase_12 AC 229 ms
77,852 KB
testcase_13 AC 171 ms
77,456 KB
testcase_14 AC 428 ms
78,648 KB
testcase_15 AC 370 ms
78,440 KB
testcase_16 AC 685 ms
80,856 KB
testcase_17 AC 771 ms
82,028 KB
testcase_18 AC 80 ms
75,240 KB
testcase_19 AC 81 ms
75,324 KB
testcase_20 AC 81 ms
75,472 KB
evil_0.txt AC 278 ms
77,800 KB
権限があれば一括ダウンロードができます

ソースコード

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