在一个由 0 和 1 组成的二维矩阵内,找到只包含 1 的最大正方形,并返回其面积。
示例:
输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
输出: 4
动态规划,记录某一点为右下角的最大正方形边长
class Solution:
def maximalSquare(self, matrix: List[List[str]]) -> int:
# dp[i][j]表示以i,j位置为右下角的最大正方形边长
if not matrix:
return 0
dp = [list(map(int, one)) for one in matrix]
x, y, res = len(dp), len(dp[0]), 0
for i in range(x):
if matrix[i][0] == '1':
res = 1
break
for j in range(y):
if matrix[0][j] == '1':
res = 1
break
for i in range(1, x):
for j in range(1, y):
if matrix[i][j] == '1':
dp[i][j] = 1 + min(dp[i-1][j], dp[i-1][j-1], dp[i][j-1])
res = max(res, dp[i][j])
return res*res