题目描述
农夫约翰的农场由 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
输出样例:
6500
算法1
(暴力枚举)
用二分法和前缀法(二分法的反馈)查找最佳牛的围栏
时间复杂度
参考文献
C++ 代码
#include<iostream>
using namespace std;
const int N=1e5+1;
int n,f;
double a[N],s[N];
bool check(double x){
for(int i=1;i<=n;i++)//表示前缀和,这里的时间复杂度为n
s[i]=s[i-1]+(a[i]-x);
double Min=0x3f3f3f3f;//将Min的初始值设定为一个非常大的数
for(int i=f;i<=n;i++){
if(s[i-f]<Min)Min=s[i-f];//找sr的最大值和sl的最小值,这里的时间复杂度为小于n
if(s[i]-Min>0)return true;//这里的s[i]表示最大值
}
return false;
}
int main(){
cin>>n>>f;
for(int i=1;i<=n;i++)cin>>a[i];
double l=1,r=2000;
while(r-l>1e-5){
double mid=(l+r)/2;//二分法,这里的时间复杂度是logn
if(check(mid)) l=mid;//这里表示二分法的判断条件
else r=mid;
}
int a=r*10000;
a/=10;
cout<<a<<endl;
return 0;
}