結果

問題 No.2204 Palindrome Splitting (No Rearrangement ver.)
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-07-28 02:11:44
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,281 bytes
コンパイル時間 508 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 564,916 KB
最終ジャッジ日時 2024-07-28 02:12:38
合計ジャッジ時間 52,274 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 TLE * 3 MLE * 1 -- * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/2204

B = 30
H = 9007199254740997

def main():
    S = input()
    # 後ろからの累積で対応
    hash_array = [0] * len(S)
    hs = 0
    for i in reversed(range(len(S))):
        hs *= B
        hs %= H
        hs += ord(S[i]) - ord("a") + 1
        hs %= H
        hash_array[i] = hs

    pow_b = [0] * (len(S) + 1)
    b = 1
    for i in range(len(S) + 1):
        pow_b[i] = b
        b *= B
        b %= H

    parindrome_map = {(i, i): 1 for i in range(len(S))}
    for i in range(len(S)):
        hs = 0
        for j in range(i, len(S)):
            hs *= B 
            hs %= H
            hs += ord(S[j]) - ord("a") + 1
            hs %= H
            # 偶数長
            k = j - i
            j_ = j + 1
            i_ = j_ + k
            if i_ < len(S):
                if i_ == len(S) - 1:
                    h = hash_array[j_]
                else:
                    h = (hash_array[j_] - (hash_array[i_ + 1] * pow_b[k + 1]) % H) % H
                if hs == h:
                    parindrome_map[(i, i_)] = (k + 1) * 2
            
            # 奇数長
            k = j - i
            j_ = j + 2
            i_ = j_ + k
            if i_ < len(S):
                if i_ == len(S) - 1:
                    h = hash_array[j_]
                else:
                    h = (hash_array[j_] - (hash_array[i_ + 1] * pow_b[k + 1]) % H) % H
                if hs == h:
                    parindrome_map[(i, i_)] = (k + 1) * 2 + 1

    def solve(S, parindrome_map, value):
        dp = [False for _ in range(len(S))]
        for i in range(len(S)):
            if (0, i) in parindrome_map and parindrome_map[(0, i)] >= value:
                dp[i] = True

            for j in range(i):
                if dp[j]:
                    if (j + 1, i) in parindrome_map and parindrome_map[(j + 1, i)] >= value:
                        dp[i] = True
        return dp[-1]

    low = 0
    high = len(S)
    while high - low > 1:
        mid = (high + low ) // 2
        if solve(S, parindrome_map, mid):
            low = mid
        else:
            high = mid
    if solve(S, parindrome_map, high):
        print(high)
    else:
        print(low)



                        









if __name__ == '__main__':
    main()
0