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

阳光男孩01

于 2024-08-27 09:16:55 发布

阅读量32

点赞数

文章标签： leetcode 矩阵算法

本文链接：https://blog.csdn.net/Pisasama/article/details/141590005

版权

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

前缀和 + 哈希表
- 见力扣363.
- 枚举上下边界转化一维前缀和，用哈希表储存出现次数

  class Solution {
  public:
      int numSubmatrixSumTarget(vector<vector<int>>& matrix, int k) {
          int ans = 0;
          int m = matrix.size(),n = matrix[0].size();
          for(int i=0;i<m;i++)
          {
              vector<int> sum(n);
              for(int j=i;j<m;j++)
              {
                  for(int c=0;c<n;c++)
                      sum[c] += matrix[j][c];
                  unordered_map<int,int> sumset;
                  sumset[0] = 1;
                  int s = 0;
                  for(int v:sum)
                  {
                      s += v;
                      if(sumset.find(s-k) != sumset.end())
                          ans += sumset[s-k];
                      sumset[s] ++;
                  }
              }
          }
          return ans;
      }
  };