【JZOJ A组】平均数

Description

给出包含一个N个整数的数组A。找出一段长度至少为K的连续序列，最大化它的平均值。
请注意：一段子序列的平均值是子序列中所有数的和除以它的长度。

Input

第一行包含两个整数N(1<=N<=300000)，K(1<=K<=N)。
第二行包含N个整数，代表数组A，1<=ai<=10^6。

Output

一行一个实数，代表最大的平均值。允许在0.001以内的绝对误差。

Sample Input

输入1：
4 1
1 2 3 4
输入2：
4 2
2 4 3 4
输入3：
6 3
7 1 2 1 3 6

Sample Output

输出1：
4.000000
输出2：
3.666666
输出3：
3.333333

Data Constraint

对于30%的数据，N<=5000。
对于100%的数据，1<=N<=300000, 1<=K<=N, 1<=Ai<=1000000。

思路

二分答案

二分一个mid，将所有a[i]-=mid，若一段长度>=k的区间和，则答案合法

所以考虑快速求这个东西

做前缀和，记录一下最大值，O（n）复杂度

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=300077;
const double e=1e-7;
int n,k,a[maxn];
double b[maxn],ans;
bool check(double x)
{
	for(int i=1; i<=n; i++) b[i]=b[i-1]+(double)a[i]-x;
	double c=-2147483647.0*2147483647;
	for(int i=n-k+1; i>=1; i--)
	{
		c=max(c,b[i+k-1]);
		if(c-b[i-1]>=0) return 1;
	}
	return 0;
}
int main()
{
	freopen("average.in","r",stdin); freopen("average.out","w",stdout);
	scanf("%d%d",&n,&k);
	for(int i=1; i<=n; i++)
	{
		scanf("%d",&a[i]);
	}
	double l=0,r=1000000;
	while(r-l>e)
	{
		double mid=(l+r)/2;
		if(check(mid)) ans=mid,l=mid;
		else r=mid;
	}
	printf("%.6lf",ans);
}