[力扣每日一题] 5-29 元素和为目标值的子矩阵

最新推荐文章于 2024-08-27 09:16:55 发布

Sharon2000s

最新推荐文章于 2024-08-27 09:16:55 发布

阅读量90

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_43439333/article/details/117392295

版权

leetcode 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

1074. 元素和为目标值的子矩阵数量
类似题

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

参考题解：

用sum前缀数组。暴力，或者用hash map优化。

class Solution {
public:
    int numSubmatrixSumTarget(vector<vector<int>>& matrix, int target) {
        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 - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + matrix[i - 1][j - 1];
            }
        }
        int ans = 0;
        for (int top = 1; top <= n; top++) {
            for (int bot = top; bot <= n; bot++) {
                int cur = 0;
                unordered_map<int, int> map;
                for (int r = 1; r <= m; r++) {
                    cur = sum[bot][r] - sum[top - 1][r];
                    if (cur == target) ans++;
                    if (map.count(cur - target)) ans += map[cur - target];
                    map[cur]++;
                }
            }
        }
        return ans;
    }
};