力扣 363. 矩形区域不超过 K 的最大数值和思维 set

最新推荐文章于 2022-02-04 11:36:09 发布

csu_xiji

最新推荐文章于 2022-02-04 11:36:09 发布

阅读量114

点赞数

分类专栏：力扣思维 STL

本文链接：https://blog.csdn.net/xiji333/article/details/116029582

版权

力扣同时被 3 个专栏收录

282 篇文章 2 订阅

订阅专栏

思维

181 篇文章 1 订阅

订阅专栏

STL

54 篇文章 0 订阅

订阅专栏

该博客介绍了如何解决LeetCode上的一个问题，即在给定的二维矩阵中找到和不超过K的最大子矩阵。作者通过动态规划和前缀和的方法，将时间复杂度降低到O(m^2nlgn)，有效地避免了暴力枚举导致的效率低下。博客内容详细阐述了算法思路，并提供了C++实现代码。

摘要由CSDN通过智能技术生成

https://leetcode-cn.com/problems/max-sum-of-rectangle-no-larger-than-k/
在这里插入图片描述
思路：暴力枚举的话是 $O(n^2m^2)$ 的，肯定过不了。考虑枚举左右边界 $l 、 r$ ，然后计算 $a[i]=\sum_{j=l}^{r}matrix[i][j]$ ，那么问题就转换为：在长度为 $n$ 的数组 $a$ 中，找到一个区间 $s 、 t$ ，使得 $a_s+…+a_t$ 最大且这个最大值 $< = k$ 。我们可以从前向后遍历数组 $a$ 并记录前缀和 $s u m$ ，那么对于任意 $sum_t$ ，符合题意的 $sum_s$ 满足 $sum_t-sum_s<=k$ ，也即 $sum_s>=sum_t-k$ ，所以我们只需要找到 $sum_t-k$ 的最小的数，用 $s e t$ 维护即可。总体复杂度 $O(m^2nlgn)$ 。

注：我的 $n 、 m$ 和题目中的 $n 、 m$ 是相反的。

class Solution {
public:
    int maxSumSubmatrix(vector<vector<int>>& matrix, int k) {
        int n=matrix.size(),m=matrix[0].size();
        vector<vector<int>> sum(n+1,vector<int>(m+1));
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                sum[i][j]=sum[i][j-1]+matrix[i-1][j-1];
        int ans=-0x3f3f3f3f;
        for(int i=1;i<=m;i++)
        {
            for(int j=i;j<=m;j++)
            {
                //[i,j]列
                set<int> s{0};
                int base=0;
                for(int t=1;t<=n;t++)
                {
                    base+=sum[t][j]-sum[t][i-1];
                    auto it=s.lower_bound(base-k);
                    if(it!=s.end())
                        ans=max(ans,base-*it);
                    s.insert(base);
                }
            }
        }
        return ans;
    }
};