leetcode 363. 矩形区域不超过 K 的最大数值和

题目：
给定一个非空二维矩阵 matrix，和一个整数 k，找到这个矩阵中矩形区域不超过 k 的最大数值和。

示例：

给定 matrix = [
[1, 0, 1],
[0, -2, 3]
]
k = 2
结果是 2，因为矩形区域 [[0, 1], [-2, 3]] 的数值和是 2，且 2 是不超过 k 的最大数字（k = 2）。

这道题是最大矩形面积问题的一个变种，我一开始用的是穷举的办法：复杂度为O（M^2*N^2）
time:886ms

class Solution {
public:
    int maxSumSubmatrix(vector<vector<int>>& matrix, int k) {
        int m = matrix.size();
        int n = matrix[0].size();
        vector<vector<int>>work(m, vector<int>(n, 0));
        work[0][0] = matrix[0][0];
        for (int i = 1; i<m; ++i) {
            work[i][0] = work[i - 1][0] + matrix[i][0];
        }
        for (int i = 1; i<n; ++i) {
            work[0][i] = work[0][i - 1] + matrix[0][i];
        }
        for (int i = 1; i<m; ++i) {
            for (int j = 1; j<n; ++j) {
                work[i][j] = matrix[i][j]+work[i - 1][j] + work[i][j - 1] - work[i - 1][j - 1];
            }
        }
        int maxArea = INT_MIN;
        for (int a = 0; a<m; ++a) {
            for (int b = a; b<m; ++b) {
                for (int c = 0; c<n; ++c) {
                    for (int d = c; d<n; ++d) {
                        int topArea = a == 0 ? 0 : work[a - 1][d];
                        int leftArea = c == 0 ? 0 : work[b][c - 1];
                        int ltArea = (a == 0 || c == 0) ? 0 : work[a - 1][c - 1];
                        int curSum = work[b][d] - leftArea - topArea + ltArea;
                        if (curSum == k)
                            return k;
                        if (curSum<k)
                            maxArea = max(maxArea, curSum);
                    }
                }
            }
        }
        return maxArea;
    }
};

然后去网上寻找了降维的方法：思路和代码如下：
乍一看题目蛮复杂的，如果把题目从二维降到一维该如何处理呢？一维数组中处理起来相对简单很多，遍历一维数组，用curSum[j]表示位置j之前所有数组元素之和，依次将curSum存入set中（用set存储便于后续二分查找），遍历到位置j时，如果j之前存在位置i满足curSum[j] - curSum[i] <= K，表明以位置j结尾的序列有满足条件不大于K的，在set中二分查找curSum[j] - K，查找到的位置即是以j结尾且相加和不大于K的最大序列的开始位置，最后比较并更新最大序列和；如果不存在满足curSum[j] - curSum[i] <= K的位置i，表明以位置j结尾的所有序列相加和均大于K。遍历整个一维数组即可得到最终结果。具体代码如下：

int maxSumSeq(vector<int>& num, int k) {
    if(num.empty()) return 0;
    int ret = INT_MIN, curSum = 0;
    set<int> sum;
    sum.insert(0);
    for(int i=0; i<num.size(); ++i) {
        curSum += num[i];
        auto it = sum.lower_bound(curSum-k);
        if(it != sum.end()) ret = max(ret, curSum-*it);
        sum.insert(curSum);
    }
    return ret;
}

在set中插入0是为了方便二分查找，同时表示下标0之前相加和是0。

一维数组问题解决了，如何扩展到二维数组呢？通过双层遍历行或列即可得到行或列所有可能的区间，例如选择列来进行遍历，针对每个列区间计算该区间中每行区间内元素之和，将列区间内每行元素作为一个整体，这样就把二维数组问题转化为一维数组了，再按照上面处理一维数组的方法即可得到最终结果。具体代码：

class Solution {
public:
    int maxSumSubmatrix(vector<vector<int>>& matrix, int k) {
        int row = matrix.size();
        if(!row) return 0;
        int col = matrix[0].size();
        if(!col) return 0;
        int ret = INT_MIN;

        for(int i=0; i<col; ++i) {
            vector<int> sum(row, 0);
            for(int j=i; j<col; ++j) {
                for(int r=0; r<row; ++r) sum[r] += matrix[r][j];
                int curSum = 0, curMax = INT_MIN;
                set<int> sumSet;
                sumSet.insert(0);
                for(int r=0; r<row; ++r) {
                    curSum += sum[r];
                    auto it = sumSet.lower_bound(curSum-k);
                    if(it != sumSet.end()) curMax = max(curMax, curSum-*it);
                    sumSet.insert(curSum);
                }
                ret = max(ret, curMax);
            }
        }

        return ret;
    }
};