算法竞赛进阶指南 最佳牛围栏

农夫约翰的农场由 NN 块田地组成，每块地里都有一定数量的牛，其数量不会少于 11 头，也不会超过 20002000 头。

约翰希望用围栏将一部分连续的田地围起来，并使得围起来的区域内每块地包含的牛的数量的平均值达到最大。

围起区域内至少需要包含 FF 块地，其中 FF 会在输入中给出。

在给定条件下，计算围起区域内每块地包含的牛的数量的平均值可能的最大值是多少。

输入格式

第一行输入整数 NN 和 FF，数据间用空格隔开。

接下来 NN 行，每行输入一个整数，第 i+1i+1 行输入的整数代表第 ii 片区域内包含的牛的数目。

输出格式

输出一个整数，表示平均值的最大值乘以 10001000 再 向下取整 之后得到的结果。

数据范围

1≤N≤1000001≤N≤100000
1≤F≤N1≤F≤N

输入样例：

10 6
6 
4
2
10
3
8
5
9
4
1

输出样例：

6500

思路如下：

 

代码如下：

#include <cmath>
#include <cstdio>
#include <iostream>

using namespace std;

const int N = 100010;

int n, F;
double s[N], a[N];

bool check(double ave)
{
    for (int i = 1; i <= n; i ++ ) s[i] = s[i - 1] + a[i] - ave;
    
    double mins = 0;
    for (int k = F; k <= n; k ++ )
    {
        mins = min(mins, s[k - F]);//记录每个最小值以免在更新k的时候要从[0, k - F]枚举一遍
        if (s[k] - mins >= 0) return true;
    }
    
    return false;
}

int main()
{
    cin >> n >> F;
    
    double l = 0, r = 0;
    for (int i = 1; i <= n; i ++ )
    {
        scanf("%lf", &a[i]);
        r = max(r, a[i]);//确定右边界，因为最大的平均值不能超过整个序列的最大数
    }
    
    while (r - l > 1e-5)//按照题意就是保留三位小数，因此一般要比答案少于1e-2级别
    {
        double mid = (l + r) / 2;//这里补充一个知识点若是整数二分用位运算的是向下取整
        if (check(mid)) l = mid;//而用(l + r) / 2是向0取整
        else r = mid;
    }
    
    printf("%d\n", (int)(r * 1000));//只能取r，因为是实数r与l存在误差不可能一定相等
                                    //比如当l = 6.4999, 而r = 6.5000是乘1000向下取整的答案不一样
    
    return 0;
}