题目描述:在一个由 ‘0’ 和 ‘1’ 组成的二维矩阵内,找到只包含 ‘1’ 的最大正方形,并返回其面积。
解题思路:如果matrix[i][j]=='0'
,那么(i, j)
处可以构成的最大正方形面积为1,否则可以构成最大面积由上面、左边和左上角可以构成的最大面积决定,边长为这三个位置面积最下的边长+1
,代码如下:
class Solution:
def maximalSquare(self, matrix: List[List[str]]) -> int:
m, n = len(matrix), len(matrix[0])
dp = [[0]*(n+1) for _ in range(m+1)]
res = 0
for i in range(1, m+1):
for j in range(1, n+1):
if matrix[i-1][j-1] == '0':
dp[i][j] = 0
else:
dp[i][j] = min(dp[i-1][j-1], dp[i][j-1], dp[i-1][j]) + 1
res = max(res, dp[i][j])
return res ** 2