結果

問題 No.1247 ブロック登り
コンテスト
ユーザー tpyneriver
提出日時 2020-10-19 19:44:18
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 2,173 bytes
コンパイル時間 220 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 528,668 KB
最終ジャッジ日時 2024-07-21 08:01:50
合計ジャッジ時間 6,730 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 1 MLE * 1 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
readline = sys.stdin.readline

N, K = map(int, readline().split())

A = list(map(int, readline().split()))

INF = 10**9+7

De = [[-INF]*(N+1) for _ in range(N)]

dpl = [[[-INF]*(N+1) for _ in range(N)] for _ in range(K)]
dpr = [[[-INF]*(N+1) for _ in range(N)] for _ in range(K)]

for i in range(N):
    dpl[0][i][i+1] = K*A[i]
    dpr[0][i][i+1] = K*A[i]


for t in range(0, K-1):
    for l in range(N):
        for r in range(l+1, min(l+t+2, N+1)):
            if dpl[t][l][r] == -INF:
                continue
            dl = dpl[t][l][r]
            dr = dpr[t][l][r]
            if t+2 < K and l+1 != r:
                dpl[t+2][l][r] = max(dpl[t+2][l][r], dl)
                dpr[t+2][l][r] = max(dpr[t+2][l][r], dr)
            
            if l != 0:
                dpl[t+1][l-1][r] = max(dpl[t+1][l-1][r], dl+(K-1-t)*A[l-1])
            if r != N:
                dpr[t+1][l][r+1] = max(dpr[t+1][l][r+1], dr+(K-1-t)*A[r])
            
            if t < t+r-l-1 < K:
                dpl[t+r-l-1][l][r] = max(dpl[t+r-l-1][l][r], dr)
                dpr[t+r-l-1][l][r] = max(dpr[t+r-l-1][l][r], dl)
            
            
            
            if l+1 != r:
                if (K-t)&1:
                    De[l][min(l+K-t, r)] = max(De[l][min(l+K-t, r)], dl)
                else:
                    De[l+1][min(l+K-t, r)] = max(De[l+1][min(l+K-t, r)], dl)

                if l <= r-K+t: 
                    De[r-K+t][r] = max(De[max(l, r-K+t)][r], dr)
                else:
                    if (r-K+t-l)%2 == 0:
                        De[l][r] = max(De[l][r], dr)
                    else:
                        De[l+1][r] = max(De[l+1][r], dr)
              
ans = [-INF]*N
for l in range(N):
    for r in range(l+1, N+1):
        ans[l] = max(ans[l], dpl[K-1][l][r])
        ans[r-1] = max(ans[r-1], dpr[K-1][l][r])

for l in range(N):
    for r in range(l+1, N+1):
        d = De[l][r]
        for i in range(l, r, 2):
            ans[i] = max(ans[i], d)


Ans = [-INF]*N

for i in range(N):
    if i > 0:
        Ans[i-1] = max(Ans[i-1], ans[i])
    if i < N-1:
        Ans[i+1] = max(Ans[i+1], ans[i])
print('\n'.join(map(str, Ans)))
0