2021牛客多校#4 J-Average

原题链接

			https://ac.nowcoder.com/acm/contest/11255/J

题目大意

给定一个$n\times m(1\le n,m\le 10^5)$的矩阵$W$，矩阵上的元素满足$W_{i,j}=a_i+b_j$，其中$a_i,b_j(1\le a_i,b_j\le 10^5)$已知。
求其平均值最大的连续子矩阵，且满足第一维度至少为$x(1\le x\le n)$，第二维度至少为$y(1\le y\le n)$。

题解

根据$W_{i,j}$的生成方式我们可知，$W_{i,1...m}=a_i+b_{1...m}$，$W_{1...n,j}=a_{1...n}+b_j$。
所以我们只需二分寻找$a$数组中长度至少为x的最大连续区间的平均值、$b$数组中长度至少为y的最大连续区间的平均值，并相加即可。

参考代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
const double eps=1e-8;
int n,m,x,y;
int v[N];
double e[N];
double w(int k,int sz)
{
	double l=0.0,r=100001.0;
	for(int i=1;i<=k;i++)scanf("%d",&v[i]);
	while(r-l>eps)
	{
		double mid=(l+r)/2.0,num,p=0.0;
		for(int i=1;i<=sz;i++)e[i]=e[i-1]+v[i]-mid;
		num=e[sz];
		for(int i=sz+1;i<=k;i++)
		{
			p=min(p,e[i-sz]);
			e[i]=e[i-1]+v[i]-mid;
			num=max(num,e[i]-p);
		}
		if(num>=0)l=mid;
		else r=mid;
	}
	return r;
}
int main()
{
	scanf("%d%d%d%d",&n,&m,&x,&y);
	printf("%.10lf\n",w(n,x)+w(m,y));
}