結果

問題 No.1247 ブロック登り
ユーザー persimmon-persimmonpersimmon-persimmon
提出日時 2021-03-04 15:45:59
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
RE  
実行時間 -
コード長 1,492 bytes
コンパイル時間 132 ms
コンパイル使用メモリ 13,184 KB
実行使用メモリ 44,620 KB
最終ジャッジ日時 2024-04-15 08:02:33
合計ジャッジ時間 18,562 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
from numba import njit
@njit
def main1(n,k,a):
  inf=10**10
  dp0=np.full((k,n,n),-inf,np.int64)
  dp1=np.full((k,n,n),-inf,np.int64)
  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=np.full(n,-inf,np.int64)
  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=np.array(list(map(int,input().split())),np.int64)
  ret1=main1(n,k,a)
  print(*ret1,sep='\n')
0