【GDOI2018】模拟B组Sabotage

Description

FJ 的死对头，FP，现在决定了去破坏FJ 的挤奶设备！
这个挤奶设备由一行N（3 <= N<= 100, 000）个挤奶机器，其中第i 个机器生产Mi 单位的牛奶（1<= Mi <= 10, 000）。FP 计划将机器连续的一块断开——从第i 个机器到第j 个机器（2<= i<= j<= N-1）；注意第一个和最后一个机器FJ 并不想要断开，因为这会让这次事件太容易被发现。FP 的目标是让剩下的机器的牛奶平均产量最小。FP 打算移去至少一台机器，即使对他来说不进行破坏更好。
幸运的是，FJ 已获悉FP 的邪恶计划，并且他想知道如果计划成功他的挤奶设备会被破坏成什么样。请帮助FJ 找到，如果FP 成功，最小的剩下挤奶机器的平均生产量。

Input

第一行：整数N。
第2 至N + 1 行：第i + 1 行包含Mi。

Output

输出单独一行一个整数——FP 能获得的最小的可能的平均数，四舍五入到小数点后3 位数字，小数点后三位数字都要输出。

Sample Input

5
5
1
7
8
2

Sample Output

2.667

样例解释：

最佳方案是移除7 和8，留下5; 1 以及2，使得平均值为8/3。

思路

题意我都琢磨了好久。。。

其实，这题是典型的平均数二分答案！

那么怎么判断呢?
最初的思路是直接枚举区间，逐个去掉，直接复杂度O(n^3log n)TLE!

怎么优化呢？当然是选择前缀和啦！！！O(n^2log n)TLE!

但其实，我们并不需要两头枚举。只需记录最小和最大。
O(n log n)

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,a[111111];
double l=0,r=0,mid=0,s[111111];
bool check(double x)
{
    for(int i=1; i<=n; i++) s[i]=(double)s[i-1]+a[i]-x;
    double mi=a[1],ma=a[2]-x;
    for(int i=2; i<=n-1; i++)
    {
        if(mi>s[i-1]) mi=(double)s[i-1];
        if(ma<s[i]-mi) ma=(double)s[i]-mi;
    }
    if(s[n]-ma<=0)return 1;else return 0;
}
int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        if(a[i]>r) r=a[i];
    }
    while(l<r)
    {
        mid=(double)(l+r)/2;
        if(r-l<0.0001) break;
        if(check(mid)) r=mid;else l=mid;
    }
    printf("%.3lf",l);
}