J. Stop Counting (最大平均前缀|后缀) (2021年度训练联盟热身训练赛第三场)

传送门

思路：

要求找到去除一个数列中间的一个连续片段后剩下数的平均值Max。

显然这个Max的最小值为0，我们利用最大平均值前缀和最大平均值后缀处理更新一下即可。

C++代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9+7;
const int  N = 2e6 + 5;

inline void read(int &x){
    char t=getchar();
    while(!isdigit(t)) t=getchar();
    for(x=t^48,t=getchar();isdigit(t);t=getchar()) x=x*10+(t^48);
}

int n;
double a[N], pre[N], rpre[N], sum, ans;

signed main()
{
//    IOS;

    cin >> n;
    for(int i = 1; i <= n; i ++){
        cin >> a[i];
        pre[i] = pre[i-1]+a[i];
        sum += a[i];
    }
    double prx = 0;
    int prpos = 0, flag = 0;
    for(int i = 1; i <= n; i ++){
        pre[i] /= i;   //最大平均值前缀数组处理
        if(pre[i] > prx){
            prx = pre[i];
            prpos = i;  //记录最大前缀左边界
        }
        if(pre[i]>0) flag = 1;
    }
    for(int i = n; i; i --){
        rpre[i] = rpre[i+1]+a[i];
    }
    double rprx = 0;
    int rprpos = 0;
    for(int i = n; i; i --){
        rpre[i] /= n-i+1;  //最大平均值后缀数组处理
        if(rpre[i]>rprx){
            rprx = rpre[i];
            rprpos = i; //记录最大后缀
        }
        if(rpre[i]>0) flag = 1;
    }
    if(!flag){
        printf("%.10f\n", 0);  //如果都为负数肯定选择什么数都不取
        return 0;
    }
    if(prpos<rprpos){ //如果最大平均值前缀的有边界在最大平均值后缀左边界的左边(不相交)
        ans = (prx*prpos+rprx*rprpos)/(prpos+rprpos);
    }
    else ans = sum/n;  //如果相交就考虑取全段的平均值
    ans = max(ans, prx);
    ans = max(ans, rprx); //别忘了和最大前缀后缀取哥max

    printf("%.10f\n", ans);

    return 0;
}