fork download
copy 
  1. #include <iostream>
  2. #include<bits/stdc++.h>
  3.  
  4. using namespace std;
  5.  
  6. int main() {
  7. 	int n;
  8. 	cin>>n;
  9. 	int k;
  10. 	cin>>k;
  11.  
  12. 	vector<int> h(n+1,0);
  13. 	for(int i=1;i<=n;i++){
  14. 		cin>>h[i];
  15. 	}
  16.  
  17. 	vector<int> dp(n+1,INT_MAX);
  18. 	dp[1]=0;
  19. 	dp[2]=abs(h[2]-h[1]);
  20. 	for(int i=3;i<=n;i++){
  21.  
  22. 		int j=1;
  23. 		while(j<=k && i-j>=1){
  24. 		dp[i] = min(dp[i] , abs(h[i-j] - h[i]) + dp[i-j]);
  25. 		j++;
  26. 		}
  27. 	}
  28.  
  29. 	cout<<dp[n];
  30.  
  31. 	return 0;
  32. }
Success #stdin #stdout 0s 5324KB
comments (?)
stdin
copy 
10 4
40 10 20 70 80 10 20 70 80 60
stdout
copy 
40
https://ideone.com/U8n5PW
language:
C++14 (gcc 8.3)
created:
1 day ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!