leetcode 363.矩形区域不超过K的最大数之和 - 枚举+前缀和

leetcode 363.矩形区域不超过K的最大数之和 - 枚举+前缀和

题干

给你一个 m x n 的矩阵 matrix 和一个整数 k ，找出并返回矩阵内部矩形区域的不超过 k 的最大数值和。
题目数据保证总会存在一个数值和不超过 k 的矩形区域。

示例 1：
输入：matrix = [[1,0,1],[0,-2,3]], k = 2
输出：2
解释：蓝色边框圈出来的矩形区域 [[0, 1], [-2, 3]] 的数值和是 2，且 2 是不超过 k 的最大数字（k = 2）。

示例 2：
输入：matrix = [[2,2,-1]], k = 3
输出：3

提示：
m == matrix.length
n == matrix[i].length
1 <= m, n <= 100
-100 <= matrix[i][j] <= 100
-105 <= k <= 105

进阶：如果行数远大于列数，该如何设计解决方案？

知识点&算法

没想出什么好办法，单纯用暴力枚举区间就要n^2*m级的时间复杂度（n为行数，m为列数），显然不太可行。
所以用前缀和进行优化，为了方便查找，我们先枚举上下界，然后前缀和储存当前上下界内行的前缀和（行元素为列上界内元素之和）。
由于用了前缀和，要求对应区间之和<=k，即找到一个子前缀和，使得当前前缀和和其差值<=k，我们设当前前缀为sum2，子前缀和为sum1，对于每个枚举到的sum2来说，即求sum2 - sum1 <= k，
也即求一个sum1满足 >= sum2 + k
因此我们对当前前缀之前的全部前缀进行搜索，来试图找到一个符合条件的子前缀，但是暴力搜索显然过于复杂。使用二叉搜索，数据又不一定满足有序这一前置条件。
所以我们用空间换时间，用一个内部有序的数据结构来维护每次枚举过程中的前缀和，在枚举到一个新的前缀和时，只要对数据结构中已有的前缀和进行二叉搜索，找到后更新答案即可。
**注意点：**如果找到的子前缀和与当前遍历的前缀和相等，在几何意义上就相当于选中了空区域，是不合法的，所以为了避免这种情况，我们在搜索完成后才将当前前缀加入数据结构。

题解

class Solution {
public:
    int maxSumSubmatrix(vector<vector<int>>& matrix, int k) {
        int n = matrix.size(), m = matrix[0].size(), res = INT_MIN;
        int linePreSum[m+5];
        for(int i = 0 ; i < n ; ++i){
            memset(linePreSum,0,sizeof(linePreSum));
            for(int j = i ; j < n ; ++j){
                int lineSum = 0;
                set<int> sum;
                sum.insert(0);
                for(int ii = 0 ; ii < m ; ++ii){
                    lineSum += matrix[j][ii];
                    linePreSum[ii+1] += lineSum;
                    auto t = sum.lower_bound(linePreSum[ii+1] - k);
                    if(t != sum.end()) res = max(res,linePreSum[ii+1] - *t);
                    sum.insert(linePreSum[ii+1]);
                }
            }
        }
        return res;
    }
};