2021牛客多校4J 二分最大区间平均数

解决矩阵中寻找平均数最大子矩阵的问题,通过将问题转化为求两个序列的最大平均数区间并结合二分法来高效求解。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

(差点WA死在这里了)
大意:给两个序列a[n],b[m],组成一个矩阵W[n][m],其中w[i][j]=a[i]+b[j],求矩阵中平均数最大的子矩阵
刚开始做的时候准备直接前缀矩阵打暴力,但是范围1e5,前缀矩阵暴力是O(n^4),铁tle。
后来看w[i][j]特殊的构造方式,于是算了一通
设最大平均数区间是a1<=i<=b1 a2<=j<=b2
由于w[i][j]=a[i]+b[j],由下图

请添加图片描述
不是很严谨,七分算三分猜。
所以问题就变成了分别求两个序列最大平均数区间,然后相加就是答案。
打了一道暴力枚举区间TLE了,然后难点就变成二分求最大区间平均数
二分一个值mid,假设它是最大区间平均数,然后让b[i]=a[i]-mid;
b[i]就是每个数对平均数大小的贡献度
然后算出b[i]的前缀和sum[i]。
当前最大区间平均数=max(之前最大区间平均数 , 当前前缀和减去之前最小的前缀和)
如果当前最大区间平均数>=0就说明条件成立

#include <iostream>
#include <cstdio>
#include <iomanip>
using namespace std;
const int N = 1e5;
int n, F;
double a[N + 5], b[N + 5], sum[N + 5];

int n1, m1, x1, y1;

bool ok(double mid) {
	for(int i=1;i<=n;i++) b[i] = a[i] - mid;
	//求出
	//当b[i]的区间和大于0时说明区间平均数大于mid
	for(int i=1;i<=n;i++) {
		sum[i] = sum[i - 1] + b[i];//前缀和
	}
	double minn = 1e10, ans = -1e12;
	//求长度>=F的所有区间的最大区间和
	
	for(int i=F;i<=n;i++) {
		minn = min(sum[i - F], minn); 
		ans = max(ans, sum[i] - minn);//当前前缀和减去前面最小的前缀和
	}
	return ans >= 0;
}

double solve() {
	double l = -1, r = 1e5, eps = 1e-10;//eps精度,本题要求10-6,这里为了保险开得大 
	while (r - l > eps) {//二分最大平均数
		double mid = (l + r) / 2;
		if (ok(mid))
			l = mid;
		else
			r = mid;
	}
	return r;

}

int main() {
	cin >> n1 >> m1 >> x1 >> y1;
	double ans1, ans2;
	for (int i = 1; i <= n1; i++) {
		cin >> a[i];
	}
	F = x1;
	n = n1;
	ans1 = solve();
	for (int i = 1; i <= m1; i++) {
		cin >> a[i];
	}
	F = y1;
	n = m1;
	ans2 = solve();
	ans1+=ans2;
	printf("%.10f",ans1);
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值