LeetCode 200 – 岛屿数量

题目描述

给你一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。

题目分析

深度优先搜索。使用数组 visited 标记被访问过的点。
遍历grid,值为 1 且没有被访问过,则认为是新岛屿,调用 DFS 对岛屿内的所有陆地做标记。

Java

boolean[][] visited;
char[][] grid;
int m, n;

public int numIslands(char[][] grid) {
    this.grid = grid;
    m = grid.length;
    n = grid[0].length;
    int count = 0;
    visited = new boolean[m][n];
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (grid[i][j] == '1' && !visited[i][j]) {
                count++;
                dfs(i, j);
            }
        }
    }
    return count;
}

private void dfs(int i, int j) {
    if (i < 0 || i >= m || j < 0 || j >= n || visited[i][j] || grid[i][j] != '1') {
        return;
    }
    visited[i][j] = true;
    dfs(i, j + 1);
    dfs(i, j - 1);
    dfs(i + 1, j);
    dfs(i - 1, j);
}

Kotlin

private lateinit var grid: Array<CharArray>
private lateinit var visited: Array<BooleanArray>

fun numIslands(grid: Array<CharArray>): Int {
    this.grid = grid
    visited = Array(grid.size) { BooleanArray(grid[0].size) }
    var count = 0
    for (i in grid.indices) {
        for (j in grid[0].indices) {
            if (grid[i][j] == '1' && !visited[i][j]) {
                dfs(i, j)
                count++
            }
        }
    }
    return count
}

private fun dfs(i: Int, j: Int) {
    if (i !in grid.indices || j !in grid[0].indices || visited[i][j] || grid[i][j] != '1') {
        return
    }
    visited[i][j] = true
    dfs(i, j + 1)
    dfs(i, j - 1)
    dfs(i - 1, j)
    dfs(i + 1, j)
}

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

开始在上面输入您的搜索词,然后按回车进行搜索。按ESC取消。

返回顶部