在本篇文章中,我们将详细解读力扣第221题"最大正方形"。通过学习本篇文章,读者将掌握如何使用多种方法来解决这一问题,并了解相关的复杂度分析和模拟面试问答。每种方法都将配以详细的解释,以便于理解。
问题描述
力扣第221题"最大正方形"描述如下:
在一个由
'0'
和'1'
组成的二维矩阵中,找到只包含'1'
的最大正方形,并返回其面积。示例:
plaintext输入: matrix = [ ["1","0","1","0","0"], ["1","0","1","1","1"], ["1","1","1","1","1"], ["1","0","0","1","0"] ] 输出: 4
示例:
plaintext输入: matrix = [ ["0","1"], ["1","0"] ] 输出: 1
解题思路
方法一:动态规划
-
初步分析:
- 使用动态规划来记录每个位置的最大正方形边长,最后返回最大边长的平方作为面积。
-
步骤:
- 定义一个二维数组
dp
,dp[i][j]
表示以matrix[i][j]
为右下角的最大正方形的边长。 - 动态转移方程:如果
matrix[i][j] == '1'
,那么dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1
。 - 遍历整个矩阵,更新
dp
数组,同时记录最大的边长。
- 定义一个二维数组
代码实现
python
def maximalSquare(matrix):
if not matrix or not matrix[0]:
return 0
m, n = len(matrix), len(matrix[0])
dp = [[0] * n for _ in range(m)]
max_side = 0
for i in range(m):
for j in range(n):
if matrix[i][j] == '1':
if i == 0 or j == 0:
dp[i][j] = 1
else:
dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1
max_side = max(max_side, dp[i][j])
return max_side * max_side
# 测试案例
print(maximalSquare([
["1","0","1","0","0"],
["1","0","1","1","1"],
["1","1","1","1","1"],
["1","0","0","1","0"]
])) # 输出: 4
print(maximalSquare([
["0","1"],
["1","0"]
])) # 输出: 1
方法二:优化的动态规划(空间优化)
-
初步分析:
- 可以将二维的
dp
数组压缩为一维数组,减少空间复杂度。
- 可以将二维的
-
步骤:
- 使用一个一维数组
dp
来记录当前行的最大正方形边长,结合一个额外变量prev
来保存左上角的值。
- 使用一个一维数组
代码实现
python
def maximalSquare(matrix):
if not matrix or not matrix[0]:
return 0
m, n = len(matrix), len(matrix[0])
dp = [0] * n
max_side = 0
prev = 0
for i in range(m):
for j in range(n):
temp = dp[j]
if matrix[i][j] == '1':
if j == 0:
dp[j] = 1
else:
dp[j] = min(dp[j], dp[j-1], prev) + 1
max_side = max(max_side, dp[j])
else:
dp[j] = 0
prev = temp
return max_side * max_side
# 测试案例
print(maximalSquare([
["1","0","1","0","0"],
["1","0","1","1","1"],
["1","1","1","1","1"],
["1","0","0","1","0"]
])) # 输出: 4
print(maximalSquare([
["0","1"],
["1","0"]
])) # 输出: 1
复杂度分析
- 时间复杂度 :O(m * n),其中 m 和 n 分别是矩阵的行数和列数。需要遍历整个矩阵以更新
dp
数组。 - 空间复杂度 :
- 二维动态规划:O(m * n),用于存储
dp
数组。 - 优化的动态规划:O(n),一维数组的大小为矩阵的列数。
- 二维动态规划:O(m * n),用于存储
模拟面试问答
问题 1:你能描述一下如何解决这个问题的思路吗?
回答 :我们可以使用动态规划来解决这个问题。通过定义一个 dp
数组,记录以每个位置为右下角的最大正方形的边长。遍历矩阵,更新 dp
数组,并记录最大的边长,最后返回其平方作为面积。
问题 2:为什么选择使用动态规划来解决这个问题?
回答 :动态规划是一种高效处理二维矩阵问题的技术,通过记录子问题的最优解,可以快速计算出全局最优解。在本题中,通过定义 dp
数组并更新每个位置的最大正方形边长,可以在 O(m * n) 的时间复杂度内解决问题,适合处理较大规模的矩阵。
问题 3:你的算法的时间复杂度和空间复杂度是多少?
回答:算法的时间复杂度是 O(m * n),其中 m 和 n 分别是矩阵的行数和列数。空间复杂度有两种情况:如果使用二维动态规划,空间复杂度为 O(m * n);如果进行空间优化,使用一维动态规划,空间复杂度可以降低到 O(n)。
问题 4:在代码中如何处理边界情况?
回答 :如果矩阵为空,或者矩阵的行或列为空,直接返回 0。此外,初始化 dp
数组时,第一行和第一列的值需要单独处理,因为它们无法从左上方元素推导出边长。通过这些边界处理,可以确保算法的正确性。
问题 5:你能解释一下动态规划在这个问题中的作用吗?
回答 :动态规划通过记录之前计算过的最优解,避免了重复计算。具体来说,通过定义 dp[i][j]
表示以 matrix[i][j]
为右下角的最大正方形的边长,可以根据左上、上、左三个方向的最优解快速计算出当前点的最优解,最终得出整个矩阵的最大正方形的面积。
问题 6:在代码中如何确保返回的结果是正确的?
回答 :通过遍历整个矩阵并更新 dp
数组,确保每个位置的最大正方形边长都被正确计算。通过记录过程中出现的最大边长,最后返回最大边长的平方作为结果。这个过程保证了结果的正确性。
问题 7:你能举例说明在面试中如何回答优化问题吗?
回答 :在面试中,如果被问到如何优化算法,我会先解释当前算法的瓶颈,比如空间复杂度。然后可以提出空间优化方案,比如将二维 dp
数组压缩为一维,降低空间复杂度。最后,通过代码实现并分析优化后的算法,解释其优缺点。
问题 8:如何验证代码的正确性?
回答:通过运行多组测试用例验证代码的正确性,特别是边界情况的测试,如矩阵为空、矩阵只有一行或一列的情况。确保每个测试用例的结果都符合预期,且算法能在规定的时间内完成计算。此外,还可以通过手工推演一些简单的例子来验证代码逻辑。
问题 9:你能解释一下解决"最大正方形"问题的重要性吗?
回答:解决"最大正方形"问题在计算机视觉、图像处理等领域具有广泛的应用。例如,在处理二值图像时,识别最大面积的目标区域是一个常见的需求。通过学习这个问题,可以帮助我们理解如何在二维矩阵中高效处理动态规划问题,提高解决类似问题的能力。
问题 10:在处理大数据集时,算法的性能如何?
回答 :算法的性能主要取决于矩阵的行数 m
和列数 n
。使用动态规划可以保证在 O(m * n) 的时间复杂度内解决问题,而通过空间优化可以将空间复杂度从 O(m * n) 降低到 O(n),在处理大规模数据时具有较好的性能表现。
总结
本文详细解读了力扣第221题"最大正方形",通过使用动态规划和空间优化的动态规划方法高效地解决了这一问题,并提供了详细的解释和模拟面试问答。希望读者通过本文的学习,能够在力扣刷题的过程中更加得心应手。