输入样例:
10 6
6
4
2
10
3
8
5
9
4
1
输出样例:
6500
解析:
1. 由题意可知答案位于 1-2000以内,所以可以二分这个区间,
2. 对于每个mid,我们要看是否存在一个区间,这个区间的平均值大于mid。如果存在返回true,否则返回false。因为如果存在,则一定存在比mid更大的答案。
3. 如果暴力枚举整个区间,则复杂度为 O(n^2logn) 会超时,所以可以双指针,将复杂度将为O(nlogn)。
4. 在check中,将区间每个值减去二分的值,这样只需要统计前缀和是否大于 0 即可,并且用 minv 存放当前已遍历过的最小的前缀和。如果存在一个前缀和减去最小前缀和大于 0 则说明一定存在一个区间的平均值大于二分的mid。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=1e5;
int n,k,a[N];
double sum[N];
bool check(double x){
for(int i=1;i<=n;i++) sum[i]=sum[i-1]+double(a[i]-x); //前缀和
double minv=0; //最小前缀和
for(int i=0,j=k;j<=n;i++,j++){
minv=min(minv,sum[i]);
if(sum[j]-minv>=0) return true;
}
return false;
}
int main(){
scanf("%d%d",&n,&k);
double l=0,r=0;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
r=max(r,double(a[i])); //找最大值,缩小二分范围
}
while(r-l>1e-5){
double mid=(l+r)/2;
if(check(mid)) l=mid;
else r=mid;
}
printf("%d",int(r*1000)); //此处应为 r 右端点,因为左端点 l 因为误差,恒小于 r
return 0;
}