1.题目
2.题目意思
找出矩阵中由“1”构成的最大正方形,返回面积
3.代码
class Solution:
def maximalSquare(self, matrix: List[List[str]]) -> int:
if not matrix or not matrix[0]:
return 0
m, n = len(matrix), len(matrix[0])
dp = [[0]*(n+1) for _ in range(m+1)]
for i in range(1, m+1):
for j in range(1, n+1):
if matrix[i-1][j-1] == '1':
dp[i][j] = min(dp[i-1][j-1], dp[i][j-1], dp[i-1][j]) + 1
return max(map(max, dp)) ** 2
思路:dp~
注意题目中的0,1是字符,很坑!
d p [ i ] [ j ] dp[i][j] dp[i][j]表示以第 i i i行第 j j j列为正方形的右下角,所能得到的最大正方形的边长。
递推公式为: d p [ i ] [ j ] = m i n ( d p [ i − 1 ] [ j − 1 ] , d p [ i ] [ j − 1 ] , d p [ i − 1 ] [ j ] ) dp[i][j]=min(dp[i-1][j-1], dp[i][j-1],dp[i-1][j]) dp[i][j]=min(dp[i−1][j−1],dp[i][j−1],dp[i−1][j])+1.
不妨先考虑一个2x2
的正方形,当右下角那个点为‘1’时,只有他的右边、左边以及左上对应的dp都是1时,才能得到一个面积为4的正方形;同理,当且仅当右边、左边以及左上对应的dp都是2时,才能得到一个3x3
的正方形。
可以参考下图对照理解。左边是输入矩阵,可以看到,最大正方形面积是9,边长是3。dp矩阵结合代码可以看出。
冲冲冲~