农夫约翰的农场由 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
解题思路:用到双指针算法,二分和前缀和。二分枚举答案,然后判断枚举的答案在题目给的数组中是否存在,在判断函数中用双指针算法来判断是否存在这么一个连续的子段。
#include<iostream>
using namespace std;
int a[100005],n,f;
double sum1[100005],sum2[100005];
bool judge(double mid){
for(int i=1;i<=n;i++)sum2[i]=sum2[i-1]+a[i]-mid;
double mini=0;
for(int j=f,i=0;j<=n;j++,i++)
{
if(sum2[i]<mini)mini=sum2[i];
if(sum2[j]-mini>=0)return true;
}
return false;
}
int main(){
scanf("%d%d",&n,&f);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
double l=0,r=2000,mid;
while(r-l>1e-5)
{
mid=(l+r)/2;
if(judge(mid)==1)l=mid;
else r=mid;
}
printf("%d",int(r*1000));
}