原题
在一个二维01矩阵中找到全为1的最大正方形
样例
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
返回 4
解题思路
- 动态规划
- dp[i][j]跟左上，左，上三个位置最大能延伸的正方形边长相关，如果dp[i][j]不等于0，则是三个中的最小值+1
- 初始化第一行，第一列，然后双层for循环更新
完整代码
class Solution(object):
def maximalSquare(self, matrix):
"""
:type matrix: List[List[str]]
:rtype: int
"""
if not matrix or not matrix[0]:
return 0
width = len(matrix[0])
height = len(matrix)
dp = [[0 for i in range(width)] for j in range(height)]
ans = 0
for i in range(width):
dp[0][i] = int(matrix[0][i])
ans = max(ans, dp[0][i])
for j in range(height):
dp[j][0] = int(matrix[j][0])
ans = max(ans, dp[j][0])
for j in range(1, height):
for i in range(1, width):
if matrix[j][i] != "0":
dp[j][i] = min(dp[j-1][i-1], min(dp[j-1][i], dp[j][i-1])) + 1
ans = max(ans, dp[j][i])
return ans * ans
