农夫约翰的农场由 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
解析:
二分答案。
一般求平均值最大。那我们可以用原数组值-平均值,然后在维护一个前缀和。如果某一段的前缀和>=0 && 长度>=f 说明符合题意,让l=mid 继续寻找最大值
某一段的前缀和>=0 就相当于 sum[j]>=sum[i] 。sum[i] 就相当于[1~j-1 ]这段区间的最小值
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10000;
double a[N],sum[N];
int n,f;
bool check(double x)
{
for(int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i]-x;
double minn=0;
for(int i=f,j=0;i<=n;i++,j++)
{
minn=min(minn,sum[j]);
if(sum[i]>=minn) return true;
}
return false;
}
int main()
{
scanf("%d %d",&n,&f);
double l=0,r;
for(int i=1;i<=n;i++)
{
cin>>a[i];
r=max(r,a[i]);
}
while(r-l>1e-5)
{
double mid=(l+r)/2;
if(check(mid)) l=mid;
else r=mid;
}
cout<<(int)(r*1000)<<endl;
}