题目
你一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
示例 1:
输入:
[
['1','1','1','1','0'],
['1','1','0','1','0'],
['1','1','0','0','0'],
['0','0','0','0','0']
]
输出: 1
示例 2:
输入:
[
['1','1','0','0','0'],
['1','1','0','0','0'],
['0','0','1','0','0'],
['0','0','0','1','1']
]
输出: 3
解释: 每座岛屿只能由水平和/或竖直方向上相邻的陆地连接而成。
思路一 BSF
采用广度优先搜索
算法思路步骤
- 首先遍历整个这个二维网格
- 遇到是‘1’的且没有被访问过的,将这个元素加入队列
- 队列出元素,将元素的上下左右四个位置,访问然后进行判断,是0就不操作,是‘1’且没有被访问过且没有超出边界,就加入到队列中,并标记为访问
- 直到队列为空,就是为一个岛屿了
public class Solution
{
private int rows;
private int cols;
public int numIslands(char[][] grid) {
// x-1,y
// x,y-1 x,y x,y+1
// x+1,y
int[][] directions = {{-1, 0}, {0, -1}, {1, 0}, {0, 1}};
rows = grid.length;
if (rows == 0) {
return 0;
}
cols = grid[0].length;
boolean[][] marked = new boolean[rows][cols]; //标记网格是否访问过
int count = 0;
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
if (!marked[i][j] && grid[i][j] == '1') {
count++;
LinkedList<Integer> queue = new LinkedList<>();
queue.addLast(i * cols + j);//记住当前格子的位置,几行几列,作为一个int存放
marked[i][j] = true;
while (!queue.isEmpty()) {
int cur = queue.removeFirst();
int curX = cur / cols;
int curY = cur % cols;
for (int k = 0; k < 4; k++) { //访问当前元素上下左右位置
int newX = curX + directions[k][0];
int newY = curY + directions[k][1];
if (inArea(newX, newY) && grid[newX][newY] == '1' && !marked[newX][newY]) {
queue.addLast(newX * cols + newY);
marked[newX][newY] = true;
}
}
}
}
}
}
return count;
}
//判断网格是否出界
private boolean inArea(int x, int y) {
return x >= 0 && x < rows && y >= 0 && y < cols;
}
}
复杂度分析
时间复杂度:O(MN),遍历M*N的二维网格
空间复杂度:O(MN),标记是否访问过的boolean数组大小为MN,队列的话最坏情况大小为 Min O(M,N)
优化 空间复杂度
这里我们额外创建了一个boolean的二维数组,用来存储网格是否为访问过,我们可以不用这个二维数组,遇到符合情况的网格,可以将它变为0,这样到最后网格都为0,岛屿数量也可以得出
思路二,DFS
采用深度优先遍历
算法思路
- 遍历二维网格数组,如果是没有访问过且是‘1’,最这个点进行dfs,岛屿数量加一
- dfs,就是将这个元素进行标记访问过,然后对左右上下四个方向进行判断, 是‘1’且没有访问过且没有出界,然后继续进行dfs
public class Solution {
// 方向数组,它表示了相对于当前位置的 4 个方向的横、纵坐标的偏移量
private static final int[][] directions = {{-1, 0}, {0, -1}, {1, 0}, {0, 1}};
// 标记数组,标记了 grid 的坐标对应的格子是否被访问过
private boolean[][] marked;
private int rows;
private int cols;
private char[][] grid;
public int numIslands(char[][] grid) {
rows = grid.length;
if (rows == 0) {
return 0;
}
cols = grid[0].length;
this.grid = grid;
marked = new boolean[rows][cols];
int count = 0;
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
// 如果是岛屿中的一个点,并且没有被访问过
// 就进行深度优先遍历
if (!marked[i][j] && grid[i][j] == '1') {
count++;
dfs(i, j);
}
}
}
return count;
}
private void dfs(int i, int j) {
marked[i][j] = true;
// 得到 4 个方向的坐标
for (int k = 0; k < 4; k++) {
int newX = i + directions[k][0];
int newY = j + directions[k][1];
// 如果不越界、没有被访问过、并且还要是陆地
if (inArea(newX, newY) && grid[newX][newY] == '1' && !marked[newX][newY]) {
dfs(newX, newY);
}
}
}
private boolean inArea(int x, int y) {
return x >= 0 && x < rows && y >= 0 && y < cols;
}
}
复杂度分析
时间复杂度:O(MN),其中 M和 N分别为行数和列数。
空间复杂度:O(MN),boolean数组大小为MN,然后在最坏情况下,整个网格均为陆地,深度优先搜索的深度达到 MN。