leetcode 1292.元素之和小于等于阈值的正方形区域的最大边长【二维前缀和+二分查找】

给你一个大小为 m x n 的矩阵 mat 和一个整数阈值 threshold。

请你返回元素总和小于或等于阈值的正方形区域的最大边长；如果没有这样的正方形区域，则返回 0 。

示例 1：

输入：mat = [[1,1,3,2,4,3,2],[1,1,3,2,4,3,2],[1,1,3,2,4,3,2]], threshold = 4
输出：2
解释：总和小于或等于 4 的正方形的最大边长为 2，如图所示。

示例 2：

输入：mat = [[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2]], thresh
old = 1
输出：0

示例 3：

输入：mat = [[1,1,1,1],[1,0,0,0],[1,0,0,0],[1,0,0,0]], threshold = 6
输出：3

示例 4：

输入：mat = [[18,70],[61,1],[25,85],[14,40],[11,96],[97,96],[63,45]], threshold = 40184
输出：2

提示：

1 <= m, n <= 300
m == mat.length
n == mat[i].length
0 <= mat[i][j] <= 10000
0 <= threshold <= 10^5

思路：
方法一：二维前缀和+二分查找
正方形子矩阵的元素和 <= 阈值的最大边长。
计算二维矩阵的前缀和，sum[i][j] = sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1] + matrix[i-1][j-1]. （对sum添加辅助下标sum[0][j]和sum[i][0])
对矩形的较小的维度二分，l,r = 1, min(len(matrix), len(matrix[0]), mid = (l+r) // 2, 边长=mid时，如果存在元素和<=阈值的正方形子矩阵，
则扩展到更大的正方形边长，l右移，否则，缩小到更小的正方形边长，r左移。

时间复杂度O(mnlog(min(m,n))

code:

class Solution:
    def maxSideLength(self, mat, threshold) -> int:
        m, n = len(mat), len(mat[0])
        ans = 0
        # 二维矩阵前缀和
        presum = [[0]*(n+1) for _ in range(m+1)]
        for i in range(1, m+1):
            for j in range(1, n+1):
                presum[i][j] = presum[i-1][j] + presum[i][j-1] - presum[i-1][j-1] + mat[i-1][j-1]
        # 二分正方形边长
        l, r = 1, min(m, n)
        # print(r)
        while l <= r:
            mid = (l + r) // 2
            flag = False
            for i in range(1, m - mid + 2):  # 枚举上边界
                for j in range(1, n - mid + 2):  # 枚举左边界
                    # 正方形子矩阵是否元素和是否 <= thrashold
                    if presum[i+mid-1][j+mid-1] - presum[i-1][j+mid-1] - presum[i+mid-1][j-1] + presum[i-1][j-1] <= threshold:
                        flag = True
            if flag:
                ans = mid
                # print(ans)
                l = mid + 1
            else:
                r = mid - 1
        return ans

s = Solution()
print(s.maxSideLength(mat = [[18,70],[61,1],[25,85],[14,40],[11,96],[97,96],[63,45]], threshold = 40184))

方法二：二维前缀和+枚举优化
计算二维矩阵前缀和，设边长ans=0，然后枚举每个元素作为左上角元素时，从边长ans+1开始，如果由该边长构成的正方形元素之和满足阈值条件，则扩大边长，否则结束该元素的边长扩长。继续从下一个元素作为左上角元素，以上一轮的边长ans+1开始作为最小边长，进行扩宽边长。
分析：从第一个元素开始枚举边长，得到符合条件的边长ans1, 共枚举ans1次。继续枚举第二个元素，ans1+1作为初始边长，如果第二个元素有满足边长为ans1+1,并且满足阈值条件，继续扩宽直到得到满足条件的最大边ans2,共枚举ans2-ans1次；如果第二个元素从边长ans1+1始就不满足阈值条件，则跳到下一个元素，共枚举0次。继续枚举第三个元素…，直到枚举完所有元素。对所有元素枚举的时间复杂度是O(mn),对每个元素枚举边长时，要么是1次，要么不超过min(m,n)。
时间复杂度O(mn)