【二分+前缀和】ACW 102. 最佳牛围栏

题目

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

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

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

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

输入格式

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

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

输出格式

输出一个整数，表示围起区域内每块地包含的牛的数量的平均值可能的最大值乘以1000得到的数值。

数据范围

1≤N≤100000

1≤F≤N

解析

本质上是实数二分题型，关键在于如何构造check函数

// 实数二分模版
bool check(double mid) {
    // 待实现
}

double bsearch(double l, double r) {
    while (r - l > 1e-5) {
        double mid = (l + r)/2;
        if (check(mid)) l = mid;
        else r = mid;
    }
    return l;
}

代码

#include <iostream>
#include <cstdlib>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
#include <cstdio>

using namespace std;
const int N = 1e5 + 100;
double cow[N];
double sum[N]; // 前缀和
int n, m;

bool check(double mid) {
    // 计算减去mid后的前缀和
    // 本质做法(sum[i+k] - sum[i])/k >= mid (k >= m)
    // 转化为 sum[i+k] - sum[i] - k * mid >= 0
    for (int i = 1; i <= n; i++) sum[i] = sum[i - 1] + (cow[i] - mid);
    double minv = 0;
    for (int i = 0, j = m; j <= n; i++, j++) {
        minv = min(minv, sum[i]); // 相当于不断递增k
        if (sum[j] >= minv) return true; // 有一个满足条件就行
    }
    return false;
}

double bsearch(double l, double r) {
    while (r - l > 1e-5) {
        double mid = (l + r) / 2;
        if (check(mid)) l = mid;
        else r = mid;
    }
    return r; // 往较大处取数
}

int main() {
    cin >> n >> m;
    double l = 0, r = 0;
    for (int i = 1; i <= n; ++i) {
        cin >> cow[i];
        r = max(r, cow[i]);
    }
    int res = int(bsearch(l, r) * 1000);
    cout << res << endl;
    return 0;
}