Submission #605372

#	Time	Username	Problem	Language	Result	Execution time	Memory
605372		lunchbox	Feast (NOI19_feast)	C++17	59 / 100	115 ms	2648 KiB

feast

#include <bits/stdc++.h>
using namespace std;

const int N = 300000;

pair<long long, int> solve(int n, long long *pp, long long t) {
  pair<long long, int> last{0, 0}, dp{0, 0};
  for (int i = 0; i < n; i++) {
    dp = max(dp, make_pair(last.first + pp[i] - t, last.second + 1));
    last = max(last, make_pair(dp.first - pp[i], dp.second));
  }
  return dp;
}

int main() {
  int n, k;
  scanf("%d%d", &n, &k);
  static long long pp[N];
  for (int i = 0; i < n; i++) {
    int a;
    scanf("%d", &a);
    pp[i] = (i > 0 ? pp[i - 1] : 0) + a;
  }
  long long low = 0, hi = 0x3f3f3f3f3f3f3f3f;
  while (low < hi) {
    long long t = (low + hi) / 2;
    if (solve(n, pp, t).second > k)
      low = t + 1;
    else
      hi = t;
  }
  printf("%lld\n", solve(n, pp, low).first + low * k);
  return 0;
}

Compilation message (stderr)

feast.cpp: In function 'int main()':
feast.cpp:17:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   17 |   scanf("%d%d", &n, &k);
      |   ~~~~~^~~~~~~~~~~~~~~~
feast.cpp:21:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   21 |     scanf("%d", &a);
      |     ~~~~~^~~~~~~~~~

• Subtask 1: 0 / 4
• Subtask 2: 0 / 8
• Subtask 3: 18 / 18
• Subtask 4: 10 / 10
• Subtask 5: 11 / 11
• Subtask 6: 20 / 20
• Subtask 7: 0 / 29