寻找段落(单调队列)

题目
思路:
直接二分找答案,复杂度(n *logn)

我们将a全部减去mid,问题转化为判断是否存在一个长度在s~t范围内的区间它的和为正,如果有说明还有更大的平均值。

用前缀和和单调队列维护。

然后用单调队列求出sum[i]-min(sum[i-t]~sum[i-s]),然后判断是否大于大于0即可。

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int head,tail,n,s,t,a[N],q[N];
double sum[N],ans,l,r,mid;
bool check(double x)
{
    
    
	for(int i=1;i<=n;i++)sum[i]=sum[i-1]+double(a[i])-x;
	head=1,tail=0;
	for(int i=s;i<=n;i++)
	{
    
    
		while(head<=tail&&sum[i-s]<sum[q[tail]])tail--;
		q[++tail]=i-s;
		while(head<=tail&&q[head]<i-t)head++;
		if(head<=tail&&sum[i]>=sum[q[head]])return 1;
	}
	return 0;
}
int main()
{
    
    
	cin>>n>>s>>t;
	for(int i=1;i<=n;i++)cin>>a[i];
	l=-10000,r=10000;
	while(r-l>1e-5)
	{
    
    
		mid=(l+r)/2;
		if(check(mid))l=ans=mid;
		else r=mid;
	}
	printf("%.3lf\n",ans);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_46001550/article/details/108731101