結果

問題 No.599 回文かい
ユーザー koyumeishi
提出日時 2017-11-07 17:00:13
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 853 ms / 4,000 ms
コード長 849 bytes
コンパイル時間 231 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 83,456 KB
最終ジャッジ日時 2024-11-24 04:49:57
合計ジャッジ時間 6,859 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

import array

def zAlgo(s) :
  n = len(s)
  # z = [0 for i in range(n)]
  z = array.array('i', [0 for i in range(n)])
  z[0] = n
  [l,r] = [0,0]
  for i in range(1,n) :
    if i > r :
      [l,r] = [i,i]
      while r<n and s[r] == s[r-l] :
        r += 1

      z[i] = r-l
    else :
      k = i-l
      if z[k] < r-i+1 :
        z[i] = z[k]
      else :
        l = i
        while r<n and s[r] == s[r-l] :
          r += 1

        z[i] = r-l

    r -= 1

  return z

mod = 10**9 + 7

s = input()

n = len(s)

# dp = [0 for i in range(n)]
dp = array.array('i', [0 for i in range(n)])
dp[0] = 1

ans = 0
for i in range(n//2) :
  m = n-i*2
  if m <= 0 :
    break

  t = s[i:i+m]
  z = zAlgo(t)
  for j in range(1, m//2+1) :
    if z[m-j] == j :
      dp[i+j] = (dp[i+j] + dp[i]) % mod


for i in range(n) :
  ans = (ans + dp[i]) % mod

print(ans)
0