fork download
  1. #include <iostream>
  2. #include <string.h>
  3. #include <algorithm>
  4.  
  5. using namespace std;
  6. long long int dp[3][10003];
  7. long long int data[10003];
  8. int main() {
  9. // your code goes here
  10. int n,m;
  11. memset(dp,0,sizeof(dp));
  12. memset(data,0,sizeof(data));
  13. cin>>n>>m;
  14. for(int i=0;i<n;i++){
  15. cin>>data[i];
  16. }
  17.  
  18. for(int i=0;i<n;i++){
  19. for(int j=max(0,i-m);j<=i;j++){
  20. dp[0][i+m]=max(dp[0][i+m],dp[1][j]+data[i]);
  21. dp[1][j+m]=max(dp[1][j+m],dp[1][j]+data[i]);
  22. }
  23. for(int j=0;j<i;j++){
  24. dp[1][i]=max(dp[1][i],dp[0][j]+data[i]);
  25. dp[0][i+m]=max(dp[0][i+m],dp[0][j]+data[i]);
  26. }
  27. dp[1][i]=max(dp[1][i],dp[0][i]+data[i]);
  28. for(int j=0;j<2;j++){
  29. dp[j][i+1]=max(dp[j][i+1],dp[j][i]);
  30. }
  31.  
  32.  
  33. for(int j=0;j<10;j++){
  34. cout<<j<<"("<<dp[0][j]<<" "<<dp[1][j]<<")";
  35. }
  36. cout<<endl;
  37. }
  38. long long int ans=0;
  39. for(int i=0;i<2*n+3;i++){
  40. for(int j=0;j<2;j++){
  41. ans=max(ans,dp[j][i]);
  42. }
  43. }
  44. cout<<ans<<endl;
  45. return 0;
  46. }
Success #stdin #stdout 0.01s 5288KB
stdin
6 3
3 7 1 5 6 4
stdout
0(0 3)1(0 3)2(0 0)3(3 3)4(0 0)5(0 0)6(0 0)7(0 0)8(0 0)9(0 0)
0(0 3)1(0 7)2(0 7)3(3 10)4(10 10)5(0 0)6(0 0)7(0 0)8(0 0)9(0 0)
0(0 3)1(0 7)2(0 7)3(3 10)4(10 10)5(8 8)6(0 0)7(0 0)8(0 0)9(0 0)
0(0 3)1(0 7)2(0 7)3(3 10)4(10 12)5(8 12)6(15 15)7(0 0)8(0 0)9(0 0)
0(0 3)1(0 7)2(0 7)3(3 10)4(10 16)5(10 16)6(15 16)7(19 19)8(0 0)9(0 0)
0(0 3)1(0 7)2(0 7)3(3 10)4(10 16)5(10 16)6(15 16)7(19 20)8(20 20)9(0 0)
20