YbtOJ——二分算法【例题3】最大均值

C. 【例题3】最大均值

题目

若一个长度不小于$L$的子段平均值为$mid$，那么该子段的每个数都减去$mid$后，子段和为0。
若一个长度不小于$L$的子段平均值小于$mid$，那么该子段的每个数都减去$mid$后，子段和为负。
若一个长度不小于$L$的子段平均值大于$mid$，那么该子段的每个数都减去$mid$后，子段和为正。
二分答案$mid$，将序列每数减去$mid$，找出前缀和求出平均值最大的子段之和，并做上面的判断。

代码

#include<iostream>
using namespace std;
const double eps=1e-5;
int n,l,a[100001]; 
void in()
{
	cin>>n>>l;
	for(int i=1;i<=n;i++) cin>>a[i];
}
bool check(double num)
{
	double b[100001],sum[100001];
	for(int i=1;i<=n;i++) b[i]=a[i]-num;//减去平均值
	for(int i=1;i<=n;i++) sum[i]=sum[i-1]+b[i];//前缀和
	double ans=-1e10,minv=1e10;
	for(int i=l;i<=n;i++)//求平均值最大的长度不小于L的子段之和（ ans）
	{
		minv=min(minv,sum[i-l]);
		ans=max(ans,sum[i]-minv);
	}
	if(ans>=0) return 1;
	else return 0;
}
void work()
{
	double l=0,r=2000;
	while(l+eps<r)//二分，记得是小数，要加 eps
	{
		double mid=(l+r)/2;
		if(check(mid)) l=mid;
		else r=mid;
	}
	cout<<int(r*1000);
}
int main()
{
	in();
	work();
}