[Swift]LeetCode221. 最大正方形 | Maximal Square

Given a 2D binary matrix filled with 0's and 1's, find the largest square containing only 1's and return its area.

Example:

Input: 

1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

Output: 4

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

示例:

404ms

 1 class Solution {
 2     func maximalSquare(_ matrix: [[Character]]) -> Int {
 3 
 4         guard matrix.count > 0, matrix[0].count > 0 else { return 0 }
 5         var m = matrix.count, n = matrix[0].count
 6         var dp = Array(repeating: Array(repeating: 0, count: n+1), count: m+1)
 7         var res = 0
 8         for i in 1...m {
 9             for j in 1...n {
10                 if matrix[i-1][j-1] == "0" { continue }
11                dp[i][j] = min(min(dp[i-1][j], dp[i][j-1]), dp[i-1][j-1]) + 1
12                 res = max(res, dp[i][j])
13             }
14         }
15         return res * res
16     }
17 }

408ms

 1 class Solution {
 2     func maximalSquare(_ matrix: [[Character]]) -> Int {
 3 
 4         guard matrix.count > 0, matrix[0].count > 0 else { return 0 }
 5         var m = matrix.count, n = matrix[0].count
 6         var dp =  Array(repeating: 0, count: n+1)
 7         var res = 0
 8         var pre = 0
 9         for i in 0..<m {
10             for j in 1...n {
11                 var t = dp[j]
12                 if matrix[i][j-1] == "0" { dp[j] = 0 }
13                 else {
14                     dp[j] = min(min(dp[j], dp[j-1]), pre) + 1
15                      res = max(res, dp[j])
16                 }
17                 pre = t
18             }
19         }
20         return res * res
21     }
22 }

416ms

 1 class Solution {
 2     func maximalSquare(_ matrix: [[Character]]) -> Int {
 3         guard !matrix.isEmpty else { return 0 }
 4         
 5         var maxLength = 0
 6         var areaArray = Array(repeating: Array(repeating: 0, count: matrix[0].count), count: matrix.count)
 7         
 8         for i in 0..<matrix.count {
 9             for j in 0..<matrix[0].count {
10                 if i == 0 || j == 0{
11                     areaArray[i][j] = matrix[i][j] == "0" ? 0 : 1
12                 }else {
13                     if matrix[i - 1][j] == "0" && matrix[i][j - 1] == "0" && matrix[i - 1][j - 1] == "0" {
14                         areaArray[i][j] = matrix[i][j] == "0" ? 0 : 1
15                     }else if matrix[i][j] == "1" {
16                         areaArray[i][j] = min(min(areaArray[i - 1][j], areaArray[i][j - 1]), areaArray[i - 1][j - 1]) + 1
17                     }
18                 }
19                 maxLength = max(maxLength, areaArray[i][j])
20             } 
21         }
22         
23         return maxLength * maxLength
24     }
25 }

420ms

 1 class Solution {
 2     func maximalSquare(_ matrix: [[Character]]) -> Int {
 3       if matrix.count == 0 {
 4         return 0
 5       }
 6 
 7       let rows = matrix.count, cols = matrix[0].count
 8       var maxsqlen = 0
 9 
10       // make a copy of matrix
11       var dp = [[Int]](repeating: ([Int](repeating: 0, count: cols + 1)), count: rows + 1)
12 
13       for i in 1...rows {
14         for j in 1...cols {
15           // if previous i and previous j is 1
16           if matrix[i - 1][j - 1] == "1" {
17             // find the minimum of [i - 1][j], [i - 1][j - 1], [i][j - 1] and add 1
18             dp[i][j] = min(dp[i - 1][j], dp[i - 1][j - 1], dp[i][j - 1]) + 1
19             maxsqlen = max(maxsqlen, dp[i][j])
20           }
21         }
22       }
23 
24       return maxsqlen * maxsqlen
25     }
26 }

436ms

 1 class Solution {
 2     func maximalSquare(_ matrix: [[Character]]) -> Int {
 3         var maxCount = 0
 4         for row in 0..<matrix.count {
 5             for col in 0..<matrix[0].count {
 6                 guard matrix[row][col] == "1" else { continue }
 7                 var currentCount = 1
 8                 var base = 1
 9                 let startRow = row
10                 let startCol = col
11                 var newRow = row + 1
12                 var newCol = col + 1
13                 boundaryCheck: while newRow < matrix.count && newCol < matrix[0].count {
14                     for nextRow in startRow...newRow {
15                         if matrix[nextRow][newCol] != "1" {
16                             break boundaryCheck
17                         }
18                     }
19 
20                     for nextCol in startCol...newCol {
21                         if matrix[newRow][nextCol] != "1" {
22                             break boundaryCheck
23                         }
24                     }
25                     base += 1
26                     currentCount = base * base
27                     newRow += 1
28                     newCol += 1
29                 }
30                 maxCount = max(maxCount, currentCount)
31             }
32         }
33         return maxCount
34     }    
35 }

[Swift]LeetCode221. 最大正方形 | Maximal Square

猜你喜欢