fork download
  1. # your code goes here
  2. import sys
  3.  
  4. def solve():
  5. input = sys.stdin.read
  6. data = input().split()
  7. if not data:
  8. return
  9.  
  10. n = int(data[0])
  11. k = int(data[1])
  12. h = [int(x) for x in data[2:]]
  13.  
  14. dp = [float('inf')] * n
  15. dp[0] = 0
  16.  
  17. for i in range(n):
  18. for j in range(1, k + 1):
  19. if i + j < n:
  20. dp[i + j] = min(dp[i + j], dp[i] + abs(h[i + j] - h[i]))
  21.  
  22. print(dp[n - 1])
  23.  
  24. if __name__ == '__main__':
  25. solve()
Success #stdin #stdout 0.07s 14048KB
stdin
Standard input is empty
stdout
Standard output is empty