CodeForces 939E Maximize!

 

【題目連接】

 

個人作法是離線進行分治。

假設選擇第$i$個數字做爲最大值，那麼比它小的部分的數字確定是某個前綴，也就是要去尋找選擇哪一個前綴使得平均值最小。

能夠發現前綴的尋找具備決策單調性，也就是說，若是選擇第$i$個數字做爲最大值的時候，前綴是選擇了$[1, p]$；那麼選擇第$i$個數字以後的數字做爲最大值的時候，前綴是選擇確定大於等於$p$位置。所以分治就能夠了。

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

const int maxn = 5e5 + 10;
long long sum[maxn];
long long a[maxn];
double mean[maxn];
double ans[maxn];
int n;
queue<int> q;

void dp(int l, int r, int dl, int dr) {
  int mid = (l + r) >> 1, dm = dl;
  double g = 2e9;
  for(int i = dl; i <= dr && i < mid; i ++) {
    double tmp = 1.0 * (sum[i] + a[mid]) / (i + 1);
    if(tmp < g) g = tmp, dm = i;
  }
  ans[mid] = g;
  if(l < mid) dp(l, mid - 1, dl, dm);
  if(r > mid) dp(mid + 1, r, dm, dr);
}

int main() {
  int Q;
  scanf("%d", &Q);
  while(Q --) {
    int op;
    scanf("%d", &op);
    if(op == 1) {
      long long x;
      scanf("%lld", &x);
      n ++;
      a[n] = x;
      sum[n] = sum[n - 1] + a[n];
    } else {
      q.push(n);
    }
  }
  dp(1, n, 1, n);
  for(int i = 1; i <= n; i ++) {
    ans[i] = 1.0 * a[i] - ans[i];
  }
  ans[1] = 0.0;
  for(int i = 2; i <= n; i ++) {
    ans[i] = max(ans[i], ans[i - 1]);
  }
  while(!q.empty()) {
    printf("%.8f\n", ans[q.front()]);
    q.pop();
  }
  return 0;
}