提出 #15679053

---

ソースコード 拡げる

import sys 
sys.setrecursionlimit(10**6)
def dp(n,k,cache,height):
  #print(cache)
  if n==0:
    return 0
  elif n in cache :
    return cache[n]
  else:
    lcl_min=float('inf')
    i=1
    while n-i>=0 and i<=k:
      lcl_min=min(lcl_min,dp(n-i,k,cache,height)+abs(height[n]-height[n-i]))
      i+=1
      
    cache[n]=lcl_min  
    return cache[n]
n,k=map(int,input().split())
h=list(map(int,input().split()))
print(dp(n-1,k,dict(),h))

提出情報

ジャッジ結果