Leetcode 3070. Count Submatrices with Top-Left Element and Sum Less Than k

本文解析了LeetCode题目3070,介绍了如何通过构建二维累积数组来高效计算矩阵中和小于k的子矩阵数量,并给出了Python代码实现,展示了优化后的执行效率和内存占用情况。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1. 解题思路

这一题就是一个二维的累积数组的问题,我们直接求一下累积数组然后找一下满足条件的矩阵就行了。

另外,由于只需要寻找和不大于k的矩阵,因此我们可以在和超过k时提前跳出来进行剪枝,从而优化执行效率。

2. 代码实现

给出python代码实现如下:

class Solution:
    def countSubmatrices(self, grid: List[List[int]], k: int) -> int:
        n, m = len(grid), len(grid[0])
        s = [[0 for _ in range(m+1)] for _ in range(n+1)]
        ans = 0
        for i in range(n):
            for j in range(m):
                s[i+1][j+1] = grid[i][j] + s[i+1][j] + s[i][j+1] - s[i][j]
                if s[i+1][j+1] <= k:
                    ans += 1
                else:
                    break
        return ans

提交代码评测得到:耗时1591ms,占用内存65.8MB。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值