(差点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;
}