【题目链接】
我的做法是离线进行分治。
假设选择第$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; }