在数组中选取连续一段元素,长度大于等于k,找出平均值最大是多少
思路:二分答案
我们在0~最大可能值中二分答案,将答案传入check函数,用check函数判断当前答案是否可以找到一段区间其平均值大于等于当前答案,如果大于等于,那么真正答案一定大于等于当前二分答案。
check函数:
先将整个数组减去二分的答案res,然后求一个前缀和,从前缀和中找是否有某个长度大于等于k的区间和大于等于0,如果大于等于0代表一定存在另外一个区间,使得答案大于此二分答案res
int n,k
double eps = 1e-10;
double a[N],fa[N],sum[N];
int check(double res) {
int i;
for(i=1;i<=n;i++)
fa[i] = a[i] - res,sum[i] = sum[i-1]+fa[i];
double num = inf;
for(i=k;i<=n;i++) {
num = min(num,sum[i-k]);
if(sum[i]-num>=0)return 1;
}
return 0;
}
int main() {
int i,j;
cin >> n >> k;
for(i=1;i<=n;i++)
scanf("%lf",&a[i]);
double l = 0,r = N;
while(r-l>eps) {
double mid = (l+r)/2;
if(check(mid))l = mid;
else r = mid;
}
printf("%.10f\n",l);
return 0;
}