给你一个由若干 0 和 1 组成的二维网格 grid,请你找出边界全部由 1 组成的最大 正方形 子网格,并返回该子网格中的元素数量。如果不存在,则返回 0
。
示例 1:
输入:grid = [[1,1,1],[1,0,1],[1,1,1]]
输出:9
示例 2:
输入:grid = [[1,1,0,0]]
输出:1
提示:
1 <= grid.length <= 100
1 <= grid[0].length <= 100
grid[i][j] 为 0 或 1
Related Topics 数组 动态规划 矩阵
思路:
方法一:前缀和+枚举优化
记行前缀和presum_row,列前缀和presum_col. 枚举以每个元素作为左上角时,边界全部为1的最大正方形边长,用行前缀和 和 列前缀和计算加权边长。
时间复杂度
方法二:前缀和+二分查找
code:
class Solution:
def judge(self, ans, i, j, row, col):
up = row[i][j+ans-1] - row[i][j-1]
down = row[i+ans-1][j+ans-1] - row[i+ans-1][