Submission #1291482

---

Source Code Expand

#include <bits/stdc++.h>

using namespace std;

typedef long long int64;
const int64 INF = 1LL << 58;


template< class T >
std::vector< int > slideMaximum(const std::vector< T > &a, int k)
{
  int n = a.size();
  std::deque< int > deq;
  std::vector< int > res;
  for(int i = 0; i < n; i++) {
    while(deq.size() && a[deq.back()] <= a[i]) deq.pop_back();
    deq.push_back(i);
    res.push_back(deq.front());
    if(i - k + 1 >= 0) {
      if(deq.front() == i - k + 1) deq.pop_front();
    }
  }
  return res;
}


int main()
{
  int64 N, M, K, A[100000];

  cin >> N >> M >> K;
  for(int i = 0; i < N; i++) cin >> A[i];
  // dp[k][idx] := dp[k-1][j(idx-j<M)]+A_idx*k

  vector< int64 > dp1(N, 0);
  for(int i = 0; i < N; i++) dp1[i] = A[i];

  for(int i = 2; i <= K; i++) {
    vector< int64 > dp2(N, 0);
    auto dp = slideMaximum(dp1, M);
    for(int j = i - 1; j < N; j++) {
      dp2[j] = max(dp2[j], dp1[dp[j - 1]] + i * A[j]);
    }
    dp1.swap(dp2);
  }
  cout << *max_element(begin(dp1), end(dp1)) << endl;
}

Submission Info

Judge Result