結果

問題 No.1247 ブロック登り
ユーザー persimmon-persimmon
提出日時 2021-03-04 14:35:59
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,744 bytes
コンパイル時間 191 ms
コンパイル使用メモリ 82,200 KB
実行使用メモリ 827,488 KB
最終ジャッジ日時 2024-10-04 21:13:22
合計ジャッジ時間 6,293 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 1 MLE * 2 -- * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main1(n,k,a):
  inf=float('inf')
  dp0=[[[-inf]*n for _ in range(n)] for _ in range(k)]
  # dp0[i][l][r]:高さi以上のブロックは構築済みで[l,r]が到達済みでlにいるときの最大得点
  dp1=[[[-inf]*n for _ in range(n)] for _ in range(k)]
  # dp1[i][l][r]:高さi以上のブロックは構築済みで[l,r]が到達済みでrにいるときの最大得点
  for i in range(n):
    if i>0:dp0[k-1][i-1][i]=k*a[i]+(k-1)*a[i-1]
    if i<n-1:dp1[k-1][i][i+1]=k*a[i]+(k-1)*a[i+1]
  ans=[-inf]*n
  for j in range(k-1,0,-1):
    for l in range(n):
      for r in range(l,n):
        if k+1-j<r+1-l:continue
        if dp0[j][l][r]!=-inf:
          # dp0[j][l][r]からの遷移。
          if j-2>=0:dp0[j-2][l][r]=max(dp0[j-2][l][r],dp0[j][l][r])
          if j-1>=0 and l>0:dp0[j-1][l-1][r]=max(dp0[j-1][l-1][r],dp0[j][l][r]+a[l-1]*(j-1))
          # l->rへの移動
          if j-(r-l)>=0:
            dp1[j-(r-l)][l][r]=max(dp1[j-(r-l)][l][r],dp0[j][l][r])
          else:
            ans[l+j]=max(ans[l+j],dp0[j][l][r])

        if dp1[j][l][r]!=-inf:
          # dp2[j][l][r]からの遷移。
          if j-2>=0:dp1[j-2][l][r]=max(dp1[j-2][l][r],dp1[j][l][r])
          if j-1>=0 and r<n-1:dp1[j-1][l][r+1]=max(dp1[j-1][l][r+1],dp1[j][l][r]+a[r+1]*(j-1))
          # l<-rへの移動
          if j-(r-l)>=0:
            dp0[j-(r-l)][l][r]=max(dp0[j-(r-l)][l][r],dp1[j][l][r])
          else:
            ans[r-j]=max(ans[r-j],dp1[j][l][r])
  for l in range(n):
    for r in range(l,n):
      ans[l]=max(ans[l],dp0[0][l][r])
      ans[r]=max(ans[r],dp1[0][l][r])
  return ans

if __name__=='__main__':
  n,k=map(int,input().split())
  a=list(map(int,input().split()))
  ret1=main1(n,k,a)
  print(*ret1,sep='\n')
0