Advertisement
Korotkodul

кузнечик

Sep 17th, 2022 (edited)
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.73 KB | None | 0 0
  1. #include <iostream>
  2. #include <cmath>
  3. #include <vector>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <string>
  7. #include <stack>
  8. #include <set>
  9. #include <map>
  10. #define pii pair <int,int>
  11. #define vec vector
  12. using namespace std;
  13. using ll = long long;
  14. using ld = long double;
  15. using db = double;
  16. void cv(vector <int> &v){
  17.     for (auto x: v) cout<<x+1<<' ';
  18.     cout<<"\n";
  19. }
  20.  
  21. void cvl(vector <ll> &v){
  22.     for (auto x: v) cout<<x<<' ';
  23.     cout<<"\n";
  24. }
  25.  
  26.  
  27. void cvv(vector <vector <int> > &v){
  28.     for (auto x: v) cv(x);
  29.     cout<<"\n";
  30. }
  31.  
  32. void cvb(vector <bool> v){
  33.     for (bool x: v) cout<<x<<' ';
  34.     cout<<"\n";
  35. }
  36.  
  37. void cvs(vector <string>  v){
  38.     for (auto a: v){
  39.         cout<<a<<"\n";
  40.     }
  41. }
  42.  
  43. void cvp(vector <pii> a){
  44.     for (auto p: a){
  45.         cout<<p.first<<' '<<p.second<<"\n";
  46.     }
  47.     cout<<"\n";
  48. }
  49.  
  50.  
  51. bool sh=0;
  52. int main()
  53. {
  54.     ios::sync_with_stdio(0);
  55.     cin.tie(0);
  56.     cout.tie(0);
  57.     int n,k; cin>>n>>k;
  58.     vector <int> a(n,0); for (int i = 1; i <= n - 2; ++i) cin>>a[i];
  59.     vector <int> dp(n, 0), pr(n,-1);
  60.     for (int i = 1; i <= min(n-1, k); ++i){
  61.         dp[i] = a[i];
  62.         pr[i] = 0;
  63.     }
  64.     for (int i = 1; i < n; ++i){
  65.         for (int j = max(0,i - k); j <= i - 1; ++j){
  66.             if (dp[i] < dp[j] + a[i]){
  67.                 dp[i] = dp[j] + a[i];
  68.                 pr[i]  = j;
  69.             }
  70.         }
  71.     }
  72.     if (sh) cout<<"res\n";
  73.     cout<<dp[n-1]<<"\n";;
  74.     vector <int> ans = {n-1};
  75.     int v = n  - 1;
  76.     if (sh){
  77.         cout<<"dp\n";
  78.         cv(dp);
  79.         cout<<"pr\n";
  80.         cv(pr);
  81.     }
  82.     while (pr[v] != 0){
  83.         v  = pr[v];
  84.         ans.push_back(v);
  85.     }
  86.     ans.push_back(0);
  87.     reverse(ans.begin(), ans.end());
  88.     cout<<ans.size()-1<<"\n";
  89.     cv(ans);
  90. }
  91. /*
  92.  
  93. 5 3
  94. 2 -3 5
  95. */
  96.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement