题目描述
农夫约翰的农场由 N
块田地组成,每块地里都有一定数量的牛,其数量不会少于 1
头,也不会超过 2000
头。
约翰希望用围栏将一部分连续的田地围起来,并使得围起来的区域内每块地包含的牛的数量的平均值达到最大。
围起区域内至少需要包含 F
块地,其中 F
会在输入中给出。
在给定条件下,计算围起区域内每块地包含的牛的数量的平均值可能的最大值是多少。
输入格式
第一行输入整数 N
和 F
,数据间用空格隔开。
接下来 N
行,每行输入一个整数,第 i+1
行输入的整数代表第 i
片区域内包含的牛的数目。
输出格式
输出一个整数,表示平均值的最大值乘以 1000
再 向下取整 之后得到的结果。
数据范围
1≤N≤100000
1≤F≤N
样例
10 6
6
4
2
10
3
8
5
9
4
1
思路
- 由求平均值的最大值可想到用二分:由最大值可分为二段,大于等于最大值不能找到一个区间,而小于等于最大值必能至少找到一个区间。所以我们在平均值的可能范围内进行二分,取中间值判断能不能找到一个区间的平均值满足大于等于中间值,然后改变边界,迭代下去。
- 怎样快速判断一个区间的平均值大于等于mid: 让区间内的值都减去mid,若减完后区间的和大于0则成立,反之不成立
- 怎样枚举出全部的区间:可以用类似于dp的方法,将区间的右端点的选择作为集合的划分,划分为依次是f,f + 1,f + 2
…n结尾的区间(不重不漏)。 - 注意区间的范围不是f,而是大于等于f:所以应判断s[k] - mins >= 0 ? (k为枚举的右端点,mins为左端点之前的前缀和的最小值)
C++ 代码
时间复杂度(O(nlogn))
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
int n,f;
double a[N],s[N];
bool check(double mid){
for(int i = 1;i <= n;i ++) s[i] = s[i - 1] + a[i] - mid;
double mins = 0;
for(int k = f;k <= n;k ++){
mins = min(mins,s[k - f]);
if(s[k] >= mins) return true;
}
return false;
}
int main()
{
double r = 0;
scanf("%d%d",&n,&f);
for(int i = 1;i <= n;i ++){
scanf("%lf",&a[i]);
r = max(r,a[i]);
}
double l = 0;
while(r - l > 1e-5){
double mid = (l + r) / 2;
if(check(mid)) l = mid;
else r = mid;
}
printf("%d",(int)(r * 1000));
return 0;
}