洛谷 P1419 寻找段落

题目描述

给定一个长度为n的序列a_i,定义a[i]为第i个元素的价值。现在需要找出序列中最有价值的“段落”。段落的定义是长度在[S,T]之间的连续序列。最有价值段落是指平均值最大的段落，

段落的平均值=段落总价值/段落长度。

输入输出格式

输入格式：

第一行一个整数n，表示序列长度。

第二行两个整数S和T，表示段落长度的范围，在[S,T]之间。

第三行到第n+2行，每行一个整数表示每个元素的价值指数。

输出格式：

一个实数，保留3位小数，表示最优段落的平均值。

输入输出样例

输入样例#1：

3
2 2
3
-1
2

输出样例#1：

1.000

说明

【数据范围】

对于30%的数据有n<=1000。

对于100%的数据有n<=100000,1<=S<=T<=n，-10000<=价值指数<=10000。

【题目来源】

tinylic改编

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

单调队列+二分+前缀和~

二分答案，每次判断是否成立；判断的u是否可行时候使用单调队列，把记录数组统一减小u，在判断是否大于0。

和上一题很像啊。

#include<cstdio>

int n,s,t,a[100001],q[100001];
double l,r,c[100001],ans;

bool findd(double u)
{
	int le=1,ri=0;
	for(int i=1;i<=n;i++) c[i]=(double)a[i]-u*i;
	for(int i=1;i<=n;i++)
	{
		if(i>=s)
		{
			while(ri>=le && c[i-s]<c[q[ri]]) ri--;
			q[++ri]=i-s;
		}
		if(ri>=le && q[le]<i-t) le++;
		if(ri>=le && c[i]>=c[q[le]]) return 1;
	}
	return 0;
}

int main()
{
	scanf("%d%d%d",&n,&s,&t);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),a[i]+=a[i-1];
	r=10000;l=-10000;
	while(r-l>1e-5)
	{
		double mid=(l+r)/2;
		if(findd(mid)) ans=mid,l=mid;
		else r=mid;
	}
	printf("%.3lf\n",ans);
	return 0;
}