Submission #1001806

---

Source Code Expand

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;

#define MAX_N 100005
#define MAX_M 100005
#define MAX_K 305

int N, M, K;
LL A[MAX_N];

void input()
{
  cin >> N >> M >> K;
  for (size_t i = 0; i < N; i++) {
    scanf("%lld", &A[i]);
  }
}

LL dp[MAX_N][MAX_K];
deque<pair<LL, int>> numList;

void init()
{
  numList.clear();
}

void pushList(LL val, int num)
{
  pair<LL, int> a = make_pair(val, num);
  while(numList.size()){
    if(numList[numList.size()-1] < a){
      numList.pop_back();
    }
    else{
      break;
    }
  }
  numList.push_back(a);
}

LL getList(int minNum){
  while(numList.size()){
    if(numList[0].second < minNum){
      numList.pop_front();
    }
    else{
      return numList[0].first;
    }
  }
  return -1;
}

LL solve()
{
  memset(dp, -1, sizeof(dp));
  for (size_t i = 0; i < N; i++) {
    dp[i][1] = A[i];
  }

  for (int k = 1; k < K; k++) {
    init();
    for (int n = 0; n < N; n++) {
      if(getList(max(0, n-M)) >= 0){
        LL tmp = getList(max(0, n-M)) + (LL)(k+1)*A[n];
        dp[n][k+1] = max(dp[n][k+1], tmp);
      }
      pushList(dp[n][k], n);
    }
  }

  LL ret = 0;
  for (size_t i = 0; i < N; i++) {
    ret = max(ret, dp[i][K]);
  }

  return ret;
}

signed main()
{
  input();

  cout << solve() << endl;

  return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘void input()’:
./Main.cpp:18:25: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     scanf("%lld", &A[i]);
                         ^

Judge Result