LeetCodeDiary

A Diary for solving LeetCode problems

View on GitHub
from sortedcontainers import SortedList
class Solution:
    def maxSumSubmatrix(self, matrix: List[List[int]], k: int) -> int:
        ans = float("-inf")
        m, n = len(matrix), len(matrix[0])

        for i in range(m):   # 枚举上边界
            total = [0] * n
            for j in range(i, m):   # 枚举下边界
                for c in range(n):
                    total[c] += matrix[j][c]   # 更新每列的元素和
                
                totalSet = SortedList([0])
                s = 0
                for v in total:
                    s += v
                    lb = totalSet.bisect_left(s - k)
                    if lb != len(totalSet):
                        ans = max(ans, s - totalSet[lb])
                    totalSet.add(s)
        return ans

队列 二分查找 困难 每日一题 👆利用二分查找和前缀和的写法

https://leetcode-cn.com/problems/max-sum-of-rectangle-no-larger-than-k/

👇python利用暴力搜索的方法会超时

class Solution:
    def maxSumSubmatrix(self, matrix: List[List[int]], k: int) -> int:
        m, n = len(matrix), len(matrix[0])
        cum = [[0] * (n + 1) for _ in range(m + 1)]
        for i in range(1,m+1):
            for j in range(1,n+1):
                cum[i][j] = cum[i-1][j] + cum[i][j-1] - cum[i-1][j-1] + matrix[i-1][j-1]
        ans = -float('inf')
        for x1 in range(1,m+1):
            for y1 in range(1,n+1):
                for x2 in range(x1,m+1):
                    for y2 in range(y1,n+1):
                        cur = cum[x2][y2] - cum[x1-1][y2] - cum[x2][y1-1] + cum[x1-1][y1-1]
                        if cur <= k:
                            ans = max(cur,ans)
        return ans