fork download
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define ll long long int
  5. #define endl "\n"
  6.  
  7. int main(){
  8. int n,k;
  9. cin>>n>>k;
  10. vector<int> steps(n+1,0);
  11. for(int i=1;i<=n;++i){
  12. cin>>steps[i];
  13. }
  14. vector<int> dp(n+1,INT_MAX);
  15.  
  16. dp[0]=0;
  17. dp[1]=0;
  18. for(int i=2;i<=n;++i){
  19. for(int j=1;j<=k;++j){
  20. if(i-j>=0)
  21. dp[i]=min(dp[i],dp[i-j]+abs(steps[i]-steps[i-j]));
  22. }
  23. // dp[i] = min((dp[i-1]+abs(steps[i]-steps[i-1])),(dp[i-2]+abs(steps[i]-steps[i-2])));
  24. }
  25. cout<<dp[n]<<endl;
  26.  
  27. }
Success #stdin #stdout 0s 5312KB
stdin
2 1
10 10
stdout
0