Advertisement
pb_jiang

CF1661D WA

Jan 23rd, 2023
81
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.83 KB | None | 0 0
  1. #include <assert.h>
  2. #include <bits/stdc++.h>
  3. using namespace std;
  4. #define dbg(...) logger(#__VA_ARGS__, __VA_ARGS__)
  5. template <typename... Args> void logger(string vars, Args &&... values)
  6. {
  7.     cerr << vars << " = ";
  8.     string delim = "";
  9.     (..., (cerr << delim << values, delim = ", "));
  10.     cerr << endl;
  11. }
  12.  
  13. template <class T> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m)); }
  14. template <class T> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n)); }
  15. template <class T, T init> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m, init)); }
  16. template <class T, T init> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n, init)); }
  17.  
  18. template <class T> using mpq = priority_queue<T, vector<T>, greater<T>>;
  19.  
  20. using ll = long long;
  21. using pii = pair<int, int>;
  22. using vl = vector<ll>;
  23. using vi = vector<int>;
  24.  
  25. int n, k;
  26. vl b;
  27.  
  28. int main(int argc, char **argv)
  29. {
  30.     cin >> n >> k;
  31.     b = vl(n);
  32.     for (auto &x : b)
  33.         cin >> x;
  34.     ll op = 0;
  35.     vl cnt(n);
  36.     ll jcntj = 0;
  37.     ll cntj = 0;
  38.     for (int i = n - 1; i >= k - 1; --i) {
  39.         ll val = b[i] - ((k + i) * cntj - jcntj);
  40.         if (val > 0) {
  41.             cnt[i] = (val + k - 1) / k;
  42.             op += cnt[i];
  43.         }
  44.         // dbg(i, val, cnt[i], op);
  45.  
  46.         int knxt = i + k - 1;
  47.         if (knxt < n) {
  48.             jcntj = jcntj - knxt * cnt[knxt];
  49.             cntj = cntj - cnt[knxt];
  50.         }
  51.         jcntj += i * cnt[i];
  52.         cntj += cnt[i];
  53.     }
  54.     ll max_op = 0;
  55.     for (int i = k - 2; i >= 0; --i) {
  56.         ll val = b[i] - ((k + i) * cntj - jcntj);
  57.         if (val > 0) {
  58.             cnt[i] = (val + i) / (i + 1);
  59.             max_op = max(max_op, cnt[i]);
  60.         }
  61.     }
  62.     // dbg(op, max_op);
  63.     cout << op + max_op << endl;
  64.     return 0;
  65. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement