原题链接: 最大子矩阵
题目输入两个数组,A和B数组,长度分别为n,m,定义一个新的矩阵,C(i,j)=A[i]B[j]。求C的一个子矩阵,该矩阵的元素之和不超过x,在这种限制下,使得该矩阵的面积最大(i和j的乘积最大)。
在这种情况下,我们可以很自然的想到枚举矩阵的长度和宽度以及起点和终点,同时还要计算该矩阵的元素和,这样的话时间复杂度高达n的四次方,显然会超时。
这里我们先对计算矩阵元素和进行一点处理,先分别计算出A和B数组的前缀和。这样有什么好处呢?我们发现,对于子矩阵C(i,j),该矩阵的元素和等于(a1+a2+…ai)(b1+b2+…bj),其实就是一个乘法分配律而已。这样计算子矩阵元素和的时候就只需要做一次乘法就可以解决了。
但是问题还没有解决,不难想到,随着子矩阵的长或者宽的增加,子矩阵的元素和一定是越来越大的,而我们要的是面积最大,那也就是说要让乘积尽可能小,那么我们就不用枚举起点和终点了,只需要枚举长和宽的大小,当然在之前我们要先计算出各个长度下,A数组和B数组前缀和的最小值。
代码如下:
#include<iostream>
using namespace std;
const int maxn = 2000 + 5;
int a[maxn], b[maxn];
int prefixa[maxn], prefixb[maxn];
const int INF = 1e8;
int main(void)