結果

問題 No.1247 ブロック登り
ユーザー 👑 SPD_9X2
提出日時 2020-10-03 17:50:24
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,839 bytes
コンパイル時間 155 ms
コンパイル使用メモリ 82,340 KB
実行使用メモリ 406,184 KB
最終ジャッジ日時 2024-07-18 04:43:46
合計ジャッジ時間 6,204 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""

https://yukicoder.me/problems/no/1246

最終的に使う範囲は区間になる
終点を決めたとする
l,rでdp?
l,r,0→l,rの区間でどこにいるか
300*300*2*300
最後をどうするかが問題
偶奇別にセグ木を持って更新?

O(N*N*K*logN) ?

====解説を見た====

区間の間の移動は関係ない
→偶数回とどまった後に端に移動するという移動に一元化可能
→始点の問題も解決できる

dp[今いる端][今いない端][高さ] = スコアの最大値
で解ける

"""

from sys import stdin

N,K = map(int,stdin.readline().split())
A = list(map(int,stdin.readline().split()))

ans = [float("-inf")] * N

dp = [[ [float("-inf")] * (K+1) for i in range(N)] for j in range(N)]
for i in range(N):
    dp[i][i][K] = K * A[i]

for k in range(K,-1,-1):

    for a in range(N):
        for b in range(N):

            if k == 0:
                ans[a] = max(dp[a][b][k] , ans[a])
                continue

            if k-2 >= 0 and a != b:
                dp[a][b][k-2] = max(dp[a][b][k-2] , dp[a][b][k])

            if a <= b:

                #端を拡張
                if a-1 >= 0:
                    dp[a-1][b][k-1] = max(dp[a-1][b][k-1] , dp[a][b][k] + (k-1)*A[a-1] )

                if a+k <= b:
                    ans[a+k] = max(ans[a+k] , dp[a][b][k])
                if b-a <= k and a != b:
                    dp[b][a][k-(b-a)] = max(dp[b][a][k-(b-a)] , dp[a][b][k])

            if a >= b:

                if a+1 < N:
                    dp[a+1][b][k-1] = max(dp[a+1][b][k-1] , dp[a][b][k] + (k-1)*A[a+1])

                if a-k >= b:
                    ans[a-k] = max(ans[a-k] , dp[a][b][k])

                if a-b <= k and a != b:
                    dp[b][a][k-(a-b)] = max(dp[b][a][k-(a-b)] , dp[a][b][k])

   
print ("\n".join(map(str,ans)))
0