给你一个大小为 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)