【补题】牛客小白月赛86

链接：https://ac.nowcoder.com/acm/contest/73450

E-可口蛋糕

这题比赛时居然没做出来，太久没打了真的变菜了T_T。

思路

首先枚举左端点，然后根据$W$的限制求解出右端点的最小值（双指针递推），记为$r$。那么此时问题转变为：以$r$为左端点的最大子段和（通过倒序前缀和预处理即可）。

代码

#include <bits/stdc++.h>

using namespace std;

int n, W, w[1000005], d[1000005];
long long s[1000005];

int main() {
  ios::sync_with_stdio(false);
  cin >> n >> W;
  for (int i = 1; i <= n; i++) {
    cin >> w[i];
  }
  for (int i = 1; i <= n; i++) {
    cin >> d[i];
  }
  for (int i = n; i >= 1; i--) {
    s[i] = max(s[i + 1], 0ll) + d[i];
  }
  int r = 0;
  long long ans = LONG_LONG_MIN, sum = 0, f = 0;
  for (int i = 1; i <= n; i++) {
    while (r <= n && sum < W) {
      r++;
      sum += w[r];
      f += d[r];
    }
    if (r > n) {
      break;
    }
    ans = max(ans, f + s[r] - d[r]);
    sum -= w[i];
    f -= d[i];
  }
  cout << ans << '\n';
  return 0;
}

F-喜欢序列

E题卡了太久，这题思路对了但是来不及写了。

思路

首先考虑原数组可以被分成若干段连续的序列，容易证明为了使价值最大，每一段都不应该再分割。

考虑差分，做完差分后根据差分数组整个序列可以被分成若干段（差分为 1）。

维护这个差分数组，每次查询本质上是$l$和$r$的两次单点修改，每次单点修改将该点原来所在的区间拆成两部分或者合并相邻区间，用set<pair<int, int>>结构来维护，每次查询利用二分查找来确定被影响的区间。

代码

#include <bits/stdc++.h>

using namespace std;

int n, m, l, r, w;
long long ans, A[200005];
set<pair<int, int>> s;

void del(const pair<int, int>& p) {
  ans -= 1ll * (p.second - p.first + 1) * (p.second - p.first + 1);
  s.erase(p);
}

void add(int x, int y) {
  ans += 1ll * (y - x + 1) * (y - x + 1);
  s.emplace(x, y);
}

void modify(int i) {
  if (i > n || w == 0) {
    return;
  }
  auto p = *(--s.lower_bound({i, INT_MAX}));
  del(p);
  A[i] += w;
  if (p.first == i) {
    if (A[i] == 1) {
      auto p2 = *(--s.lower_bound({i, i}));
      del(p2);
      add(p2.first, p.second);
    } else {
      add(p.first, p.second);
    }
  } else {
    add(p.first, i - 1);
    add(i, p.second);
  }
}

int main() {
  ios::sync_with_stdio(false);
  cin >> n >> m;
  for (int i = 1; i <= n; i++) {
    cin >> A[i];
  }
  A[0] = 1e18;
  for (int i = n; i >= 1; i--) {
    A[i] -= A[i - 1];
  }
  r = 1;
  for (int i = 2; i <= n + 1; i++) {
    if (A[i] == 1) {
      continue;
    }
    add(r, i - 1);
    r = i;
  }
  while (m--) {
    cin >> l >> r >> w;
    modify(l);
    w = -w;
    modify(r + 1);
    cout << ans << '\n';
  }
  return 0;
}