1074.元素和为目标值的子矩阵数量(dp)

链接:1074. 元素和为目标值的子矩阵数量 - 力扣(LeetCode)

题解:

1、求前缀[0,0] 到[i,j]访问的的矩阵和

2、枚举上边界和下边界

3、右边界,逐渐扩大,通过统计前缀和的方式来求解,变为一维区间和

class Solution {
public:
    int numSubmatrixSumTarget(vector<vector<int>>& matrix, int target) {
        int m = matrix.size();
        if (m <= 0) {
            return 0;
        }
        int n = matrix[0].size();
        if (n <= 0) {
            return 0;
        }
        vector<vector<int>> prefix(m+1, vector<int>(n+1, 0));
        for (int i = 0; i < m; ++i) {
            for (int j = 0; j < n; ++j) {
                prefix[i+1][j+1] = prefix[i+1][j] + prefix[i][j+1] - prefix[i][j] + matrix[i][j];
            }
        }
        int result = 0;
        for (int t = 0; t < m; ++t) {
            for (int b = t; b < m; ++b) {
                unordered_map<int, int> prefix_sum;
                prefix_sum[0] = 1;
                for (int i = 0; i < n; ++i) {
                    int tmp = prefix[b+1][i+1] - prefix[t][i+1];
                    if (prefix_sum.find(tmp-target) != prefix_sum.end()) {
                        result += prefix_sum[tmp-target];
                    }
                    ++prefix_sum[tmp];
                }
            }
        }
        return result;
    }
};

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值