[CF] B. Array Walk - Educational Codeforces Round 92 (Rated for Div. 2)

https://codeforces.com/contest/1389/problem/B

题目大意

给出 $n$ 长的数组 $a$,可以用下述操作走 $k$ 步,其中最多 $z$ 步向左,问最大可以取到的权值。

每次操作可以进行两种中的一种:

  1. 向右走。现在在 $x$ 且 $x < n$,权值增加 $a[x + 1]$ 之后走到 $x + 1$。
  2. 向左走。现在在 $x$ 且 $x > 1$,权值增加 $a[x - 1]$ 之后走到 $x - 1$。

$2 \le n \le 10^5$。$1 \le k \le n - 1$。$0 \le z \le min(5, k)$。

$1 \le a_i \le 10^4$。

简要题解

任何一种情况如果向左走的步数定了,那么向右走的步数就也定了,也就是说最终停的位置也确定了。

假设最终停下的位置为 $x = k - left - left$ 则显然 $sum[1, x]$ 是都要取到的,之后考虑,我们总可以在向左走之前多向右走几步,即我们也有可能走到 $[x + 1, x + z]$ 这一段,这些情况未来我们必定要消耗一些步数走回到 $x$。对于没有用来往远走的“向左走”,我们可以用来在一组能到的最大的值上震荡。

注意到 $z$ 很小。所以后续枚举 $z$ 是可行的。于是我们总是枚举最远可达的位置,再在这个基础上做其他运算。

注意用尽量多的左移并不一定是最好的,因为这样也会浪费右移的次数。

复杂度

$T$:$O(n + z^2)$

$S$:$O(n)$

代码实现

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

int io_=[](){ ios::sync_with_stdio(false); cin.tie(nullptr); return 0; }();

using LL = long long;
using ULL = unsigned long long;
using LD = long double;
using PII = pair<int, int>;
using VI = vector<int>;
using MII = map<int, int>;

template<typename T> void cmin(T &x,const T &y) { if(y<x) x=y; }
template<typename T> void cmax(T &x,const T &y) { if(x<y) x=y; }
template<typename T> bool ckmin(T &x,const T &y) { 
    return y<x ? (x=y, true) : false; }
template<typename T> bool ckmax(T &x,const T &y) { 
    return x<y ? (x=y, true) : false; }
template<typename T> void cmin(T &x,T &y,const T &z) {// x<=y<=z 
    if(z<x) { y=x; x=z; } else if(z<y) y=z; }
template<typename T> void cmax(T &x,T &y,const T &z) {// x>=y>=z
    if(x<z) { y=x; x=z; } else if(y<z) y=z; }

// mt19937 rnd(chrono::system_clock::now().time_since_epoch().count());
// mt19937_64 rnd_64(chrono::system_clock::now().time_since_epoch().count());

/*
---------1---------2---------3---------4---------5---------6---------7---------
1234567890123456789012345678901234567890123456789012345678901234567890123456789
*/

void solve() {
  int n, k, z; cin >> n >> k >> z;
  vector<int> a(n);
  for (int& i : a) cin >> i;

  vector<int> sum(n);
  vector<int> mx(n + 1);
  sum[0] = a[0];
  for (int i = 1; i < n; i++) {
    sum[i] = sum[i - 1] + a[i];
    mx[i] = max(mx[i - 1], a[i] + a[i - 1]);
  }

  int ans = 0;
  for (int i = 0; i <= z; i++) {
    int right = k - i;
    int lst = right - i;
    if (lst < 0) break;
    int nans = sum[lst] + mx[lst + 1] * i;

    // cerr << lst << ' ' << nans << endl;
    cmax(ans, nans);

    int cur = sum[lst];
    for (int j = 1; j <= i; j++) {
      cur += a[lst + j - 1] + a[lst + j];
      // cerr << "  " << j << ' ' << (cur + mx[lst + j + 1] * (i - j)) << endl;
      cmax(ans, cur + mx[lst + j + 1] * (i - j));
    }
  }

  cout << ans << '\n';
}

int main() {
  int t = 1; 
  cin >> t;
  while (t--) {
    solve();
  }
  return 0;
}
Prev: [CF] A. LCM Problem - Educational Codeforces Round 92 (Rated for Div. 2)
Next: [CF] C. Good String - Educational Codeforces Round 92 (Rated for Div. 2)