fork download
  1. // written by Eng. Abdullah Haydari
  2. #include <bits/stdc++.h>
  3. using namespace std;
  4. int main(){
  5. int n,k; cin>>n>>k;
  6. int h[n],memo[n];
  7. memo[n-1]=0;
  8. for (int i=0;i<n;i++)
  9. cin>>h[i];
  10. for (int i=n-2;i>=0;i--){
  11. memo[i]=2e9;
  12. for (int j=1;j<=k and i+j<n;j++){
  13. int cost=memo[i+j]+abs(h[i]-h[i+j]);
  14. if (cost<memo[i])
  15. memo[i]=cost;
  16. }
  17. }
  18. cout<<memo[0]<<endl;
  19. }
Success #stdin #stdout 0.01s 5276KB
stdin
5 3
10 30 40 50 20
stdout
30