LeetCode - #221 最大正方形（Top 100）

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第24天，点击查看活动详情

前言

本题为 LeetCode 前 100 高频题

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新到 219 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：中等

1. 描述

在一个由 '0' 和 '1' 组成的二维矩阵内，找到只包含 '1' 的最大正方形，并返回其面积。

2. 示例

示例 1

输入：matrix = [["1","0","1","0","0"],["1","0","1","1","1"],["1","1","1","1","1"],["1","0","0","1","0"]]
输出：4

示例 2

输入：matrix = [["0","1"],["1","0"]]
输出：1

示例 3

输入：matrix = [["0"]]
输出：0

提示：

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 300
• matrix[i][j] 为 '0' 或 '1'

3. 答案

class MaximalSquare {
    func maximalSquare(matrix: [[Character]]) -> Int {
        guard matrix.count != 0 else {
            return 0
        }
    
        let m = matrix.count
        let n = matrix[0].count
    
        var max_global = 0
        var maxSquareSide = Array(count: m, repeatedValue: (Array(count: n, repeatedValue: 0)))
        
        for i in 0..<m {
            for j in 0..<n {
                guard matrix[i][j] != "0" else {
                    continue;
                }
            
                if i == 0 || j == 0 {
                    maxSquareSide[i][j] = 1
                } else {
                    maxSquareSide[i][j] = min(maxSquareSide[i][j - 1], maxSquareSide[i - 1][j], maxSquareSide[i - 1][j - 1]) + 1
                }
                
                max_global = max(max_global, maxSquareSide[i][j])
            }
        }
        
        return max_global * max_global
    }
}

• 主要思想：动态规划，新的一个等于最小平方宽度加 1。
• 时间复杂度：O(mn)
• 空间复杂度：O(mn)

该算法题解的仓库：LeetCode-Swift

点击前往 LeetCode 练习