221. Maximal Square
💡
這個題目有動態規劃的最佳解,不過這個題目如果使用窮舉的話是不會超時的,所以我很建議可以就從窮舉來想,再看看能不能優化。
窮舉法
題目給定的矩陣長寬是不定的,而假設今天如果題目的矩陣全部都是 1 ,那今天在這個矩陣長方形內,最大的正方形邊長一定是長或寬的短邊。
rows = len(matrix)
columns = len(matrix[0])
m = min(rows, columns)
這時候可以從兩個方向來寫
- 從最小的正方形邊長 1 開始檢查,一直檢查到最大的正方形邊長
m
,如果說一直到m
都可以的話,那最大面積就會是 \(m^{2}\)。 - 從最大的正方形邊長
m
開始檢查,一直檢查到最小的正方形邊長 1,如果存在一個正方形滿足題目條件,則馬上回傳面積。
我選擇的是第二條路,因為是窮舉法,又是要找最大,我當然是趕快找到最大的正方形就好。
class Solution:
def maximalSquare(self, matrix: List[List[str]]) -> int:
rows = len(matrix)
columns = len(matrix[0])
L = min(rows, columns) # Longer Side
def is_square(row, col, length):
i = row
while i < (row + length):
j = col
while j < (col + length):
if matrix[i][j] == '0':
return False
j += 1
i += 1
return True
while L > 0:
i = 0
while i < rows - L + 1:
j = 0
while j < columns - L + 1:
if is_square(i, j, L):
return L ** 2
j += 1
i += 1
L -= 1
return 0
動態規劃
如果從左上往右下開始找尋最大的正方形,一定是首先自己的位置要是 1 ,接著,去看可以來到這個位置的三個方向:上、左、左上去看看是不是都是 1 ,但是這樣我們還要繼續往回找,看看是不是都是 1 ,因此最好的方法應該是在找尋的時候,就順便記錄起來。
至於紀錄的方式,就是我們看左上、上、左,可以形成的最大正方形中,哪一個最小,動態轉移方程式就是:
dp[row][col] = min(dp[row-1][col], dp[row][col-1], dp[row-1][col-1]) + 1
class Solution:
def maximalSquare(self, matrix: List[List[str]]) -> int:
rows = len(matrix)
columns = len(matrix[0])
dp = [[0] * (columns + 1) for _ in range(rows + 1)]
m = 0
for row in range(1, rows + 1):
for col in range(1, columns + 1):
if matrix[row-1][col-1] == '1':
dp[row][col] = min(dp[row-1][col], dp[row][col-1], dp[row-1][col-1]) + 1
m = max(dp[row][col], m)
return m ** 2