Submission #3139657

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
using lint = long long int;
template<class T = int> using V = vector<T>;
template<class T = int> using VV = V< V<T> >;
template<class T> void assign(V<T>& v, int n, const T& a = T()) { v.assign(n, a); }
template<class T, class... U> void assign(V<T>& v, int n, const U&... u) { v.resize(n); for (auto&& i : v) assign(i, u...); }

int main() {
  cin.tie(NULL); ios::sync_with_stdio(false);
  int n, m, k; cin >> n >> m >> k;
  V<lint> a(n); for (int i = 0; i < n; i++) cin >> a[i];
  VV<lint> dp; assign(dp, 2, n + 1);
  for (int i = 1; i < k + 1; i++) {
    deque<int> dq;
    for (int j = max(i - m, 0); j < i; j++) {
      while(!dq.empty() and dp[i - 1 & 1][dq.back()] <= dp[i - 1 & 1][j]) dq.pop_back();
      dq.push_back(j);
    }
    for (int j = i; j < n + 1; j++) {
      dp[i & 1][j] = i * a[j - 1] + dp[i - 1 & 1][dq.front()];
      if (dq.front() == j - m) dq.pop_front();
      while (!dq.empty() and dp[i - 1 & 1][dq.back()] <= dp[i - 1 & 1][j]) dq.pop_back();
      dq.push_back(j);
    }
  }
  cout << *max_element(dp[k & 1].begin(), dp[k & 1].end()) << '\n';
}

Submission Info

Judge Result