LeetCode 363. 矩形区域不超过 K 的最大数值和--前缀和+set查询

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

题解

和这个题目很像，LeetCode 1314. 矩阵区域和–前缀和。

可以得到由左边延申到右边的矩形区域和，然后减去其中的某个符合条件的前缀和即可，然后利用set容器辅助查询。

AC代码

class Solution {
public:
    int col[105][105];//累计列方向上的前缀和
    int maxSumSubmatrix(vector<vector<int>>& matrix, int k) {
        for(int i=0;i<matrix.size();i++)
        {
            for(int j=0;j<matrix[0].size();j++)
            {
                col[i+1][j+1]=col[i][j+1]+matrix[i][j];
            }
        }
        int res=-1e9;
        set<int>q;
        for(int i1=1;i1<=matrix.size();i1++)
        {
            for(int i2=i1;i2<=matrix.size();i2++)
            {   
                q.clear();
                set<int>::iterator it;
                q.insert(0);
                int ans=0;
                for(int j=1;j<=matrix[0].size();j++)
                {
                    ans+=(col[i2][j]-col[i1-1][j]);//得到从左延申到右边的矩阵和
                    //ans-q[x]<=k -> ans-k<=q[x]
                    it=q.lower_bound(ans-k);//找到第一个大于等于ans-k的元素
                    if(it!=q.end())//说明找到了
                    {
                        res=max(res,ans-*it);
                    }
                    q.insert(ans);//插入前面的前缀和
                }
            }
        }
        return res;
    }
};