結果

問題 No.599 回文かい
ユーザー titiatitia
提出日時 2024-06-11 03:47:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,009 ms / 4,000 ms
コード長 960 bytes
コンパイル時間 250 ms
コンパイル使用メモリ 82,656 KB
実行使用メモリ 100,308 KB
最終ジャッジ日時 2024-06-11 03:48:06
合計ジャッジ時間 5,788 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,224 KB
testcase_01 AC 35 ms
52,096 KB
testcase_02 AC 34 ms
52,608 KB
testcase_03 AC 36 ms
52,224 KB
testcase_04 AC 55 ms
65,408 KB
testcase_05 AC 54 ms
65,280 KB
testcase_06 AC 55 ms
66,944 KB
testcase_07 AC 55 ms
66,688 KB
testcase_08 AC 55 ms
66,560 KB
testcase_09 AC 59 ms
66,944 KB
testcase_10 AC 234 ms
77,944 KB
testcase_11 AC 182 ms
76,924 KB
testcase_12 AC 249 ms
78,188 KB
testcase_13 AC 174 ms
77,032 KB
testcase_14 AC 545 ms
81,536 KB
testcase_15 AC 458 ms
81,044 KB
testcase_16 AC 914 ms
95,616 KB
testcase_17 AC 1,009 ms
100,308 KB
testcase_18 AC 43 ms
59,264 KB
testcase_19 AC 44 ms
59,648 KB
testcase_20 AC 49 ms
59,648 KB
evil_0.txt AC 310 ms
79,232 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

# Z algorithm
# https://snuke.hatenablog.com/entry/2014/12/03/214243
# 「S と S[i:|S|-1] の最長共通接頭辞の長さ」を記録した配列 A を構築

def z_algo(S):
    if len(S)==0:
        return []
    LEN=len(S)
    i=1
    j=0
    A=[0]*LEN
    A[0]=LEN

    while i<LEN:
        while i+j<LEN and S[j]==S[i+j]:
            j+=1
        A[i]=j
        
        if j==0:
            i+=1
            continue
        
        k=1
        while i+k<LEN and k+A[k]<j:
            A[i+k]=A[k]
            k+=1
        i+=k
        j-=k

    return A

mod=10**9+7

S=input().strip()
DP=[0]*((len(S)+1)//2+1)
DP[0]=1
ANS=0

for i in range((len(S))//2+1):
    A=z_algo(S[i:len(S)-i])
    #print(A)
    LIST=[]

    for j in range(len(A)-1):
        if A[len(A)-1-j]==j+1:
            LIST.append(j+1)
            #print("!",j+1)

    for l in LIST:
        if i+l<len(DP):
            DP[i+l]=(DP[i+l]+DP[i])%mod

    ANS=(ANS+DP[i])%mod

print(ANS)
0