LeetCode 1074. 元素和为目标值的子矩阵数量

最新推荐文章于 2024-10-29 16:28:27 发布

web18484626332

最新推荐文章于 2024-10-29 16:28:27 发布

阅读量81

点赞数

分类专栏：面试学习路线阿里巴巴文章标签： java java进阶后端

本文链接：https://blog.csdn.net/web18484626332/article/details/124776434

版权

面试同时被 3 个专栏收录

147 篇文章 7 订阅

订阅专栏

阿里巴巴

138 篇文章 0 订阅

订阅专栏

学习路线

135 篇文章 0 订阅

订阅专栏

这篇博客讨论了一种算法，通过构造二维前缀和和使用哈希表来高效解决寻找和为目标值的子矩阵数量的问题。作者指出，对于区间查询，暴力搜索是不可行的，而应该利用哈希表配合前缀和记录。文章提供了相应的代码实现，并通过遍历矩阵的上、下边界来更新哈希表，从而快速查找满足条件的子矩阵。

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

基本思想：

和之前一道题思路很想，构造二维前缀和，之后固定上下边界，从左至右进行区间计算；

可以把问题转化为：“560. 和为K的子数组“，来进行计算；

这里可以用hash+前缀求解，但是自己当时忘了；

求区间问题爆搜不可取，还是hash+前缀记录吧；

基本代码：

class Solution {
public:
    int numSubmatrixSumTarget(vector<vector<int>>& matrix, int target) {
        int m=matrix.size();
        int n=matrix[0].size();
        int ret=0;
        //unordered_map<int, vector<int>>ump;
        vector<vector<int>>dp(m+1,vector<int>(n+1,0));
        for(int i=1;i<=m;i++){
            for(int j=1;j<=n;j++){
                dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+matrix[i-1][j-1];
                //int index=(i-1)*n+j;
                //ump[dp[i][j]].push_back(index);
            }
        }
        for(int top=0;top<=m;top++){
            for(int bot=top+1;bot<=m;bot++){
                unordered_map<int, int>ump;
                ump[0]=1;
                int pre=0;
                for(int i=1;i<=n;i++){
                    int remain=dp[bot][i]-dp[top][i];
                    if(ump.count(remain-target)){
                        ret+=ump[remain-target];
                    }
                    ump[remain]++;
                }
            }
        }
        return ret;
    }
};