Swift - LeetCode - 岛屿的周长

73 阅读2分钟

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

题目

给定一个 row x col 的二维网格地图 grid,其中:grid[i][j] = 1 表示陆地,grid[i][j] = 0 表示水域。

网格中的格子 水平和垂直 方向相连(对角线方向不相连)。整个网格被水完全包围,但其中恰好有一个岛屿(或者说,一个或多个表示陆地的格子相连组成的岛屿)。

岛屿中没有“湖”(“湖” 指水域在岛屿内部且不和岛屿周围的水相连)。格子是边长为 1 的正方形。网格为长方形,且宽度和高度均不超过 100 。计算这个岛屿的周长。

示例 1:

image.png

  • 输入: grid = [[0,1,0,0],[1,1,1,0],[0,1,0,0],[1,1,0,0]]
  • 输出: 16
  • 解释: 它的周长是上面图片中的 16 个黄色的边

示例 2:

  • 输入: grid = [[1]]
  • 输出: 4

示例 3:

  • 输入: grid = [[1,0]]
  • 输出: 4

方法一:迭代

思路及解法

对于一个陆地格子的每条边,它被算作岛屿的周长当且仅当这条边为网格的边界或者相邻的另一个格子为水域。 因此,我们可以遍历每个陆地格子,看其四个方向是否为边界或者水域,如果是,将这条边的贡献(即 1)加入答案 ans\textit{ans} 中即可。

代码

class Solution {
    func islandPerimeter(_ grid: [[Int]]) -> Int {
        let dx: [Int] = [0, 1, 0, -1]
        let dy: [Int] = [1, 0, -1, 0]
        let n: Int = grid.count
        let m: Int = grid[0].count
        var ans: Int = 0
        for i in 0..<n {
            for j in 0..<m {
                if grid[i][j] > 0 {
                    var cnt = 0
                    for k in 0..<4 {
                        let tx = i + dx[k]
                        let ty = j + dy[k]
                        if (tx < 0 || tx >= n || ty < 0 || ty >= m || grid[tx][ty] < 1) {
                            cnt += 1
                        }
                    }
                    ans += cnt
                }
            }
        }
        return ans
    }
}

复杂度分析

  • 时间复杂度:O(nm)O(nm),其中 nn 为网格的高度,mm 为网格的宽度。我们需要遍历每个格子,每个格子要看其周围 4 个格子是否为岛屿,因此总时间复杂度为 O(4nm)=O(nm)O(4nm)=O(nm)

  • 空间复杂度:O(1)O(1)。只需要常数空间存放若干变量。