LeetCode 岛屿数量问题 思路整理

题目

给定一个由‘1’（陆地）和‘0’（水）组成的二维网格，计算岛屿的数量。一个岛被水包围，并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。

代码框架：

class Solution {
public:
    int numIslands(vector<vector<char>>& grid) {}
};

思路

毫无疑问，需要遍历 grid 的每个元素，并且需要将遍历到的每一个‘1’作为起点开始搜索其上、下、左、右四个方向上是否为‘1’。

1. 如果与该点邻接的上、下、左、右中的一点为‘0’，则停止该方向上的搜索探索；
2. 若临界点中有一个点为‘1’，则需要以该邻接点为起点继续探索它的邻接点的情况。

每遍历到一个‘1’，则会将包含这个‘1’的整个岛屿找出，计数+1。为了不重复遍历，需要将作为起点的‘1’改写为‘0’。

为了满足上述思路，需要对于以‘1’为起点探索邻接点的方法进行具体化设计。于是很容易可以想到BFS（广度优先搜索）和DFS（深度优先搜索）。

DFS 深度优先搜索

DFS搜索是沿某一方向搜索完全部的结果，再转向另一个方向。
遍历 grid 时，如果当前元素 grid[i][j] == ‘1’ ，则按照程序设定的第一个方向进行搜索，一直到头；再从 (i，j) 的位置沿设定的第二个方向进行搜索，一直到头…
以此类推，完成 (i，j) 周围4个方向的搜索。

class Solution {
public:
	void dfs(vector<vector<char>>& grid, int r, int c) //构造dfs函数
	{
    	int nr = grid.size();
    	int nc = grid[0].size();

    	grid[r][c] = '0';
    	if (r - 1 >= 0 && grid[r-1][c] == '1') //如果当前位置上方的元素没超边界，且为陆地
    		dfs(grid, r - 1, c);  //up           以此元素为中心开始DFS搜索
    	if (r + 1 < nr && grid[r+1][c] == '1') 
    		dfs(grid, r + 1, c);  //down
    	if (c - 1 >= 0 && grid[r][c-1] == '1') 
    		dfs(grid, r, c - 1);  //left
    	if (c + 1 < nc && grid[r][c+1] == '1') 
    		dfs(grid, r, c + 1);  //right
  	}

    int numIslands(vector<vector<char>>& grid) 
    {
		int nr = grid.size();
    	if (!nr) return 0;
    	int nc = grid[0].size();
    	int num_islands = 0;   

		//遍历每一个元素
    	for (int r = 0; r < nr; ++r) 
    	{
      		for (int c = 0; c < nc; ++c) 
      		{
        		if (grid[r][c] == '1') {
          		++num_islands;
          		dfs(grid, r, c);   //每调用一次，即找到了一个岛屿
        	}
      	}
    }
    return num_islands;
	}
};

BFS 广度优先搜索

从‘1’的位置按等距离逐层向外搜索，将搜索到的‘1’全部改为‘0’。
将遍历到的‘1’上下左右四个方向上的邻接点先按一定的顺序判断一遍是否为‘1’。如果是‘1’，记录下此点的位置，存入队列 neighbors 。
当四个方向的邻接点都判断完成的时候，按上下左右的扫描顺序，找到最早判断为‘1’的点（举例：上方向和下方向为‘1’，其余为‘0’）。以上方向的邻接点作为起点，继续判断它的直接邻接点。完成后，判断下方向的直接邻接点情况。
对于如何确定搜索的先后顺序，正是BFS的关键所在。将‘1’邻接点的vector坐标入队。将队头出队后将其‘1’邻接点入队，循环判断。
按照这种方法，遍历完全部的‘1’，在vector中每发现一个‘1’，计数增加1，即可得到岛屿的数目。编写程序如下：

class Solution {
public:
  int numIslands(vector<vector<char>>& grid) {
    int nr = grid.size();
    if (!nr) return 0;
    int nc = grid[0].size();
    int num_islands = 0;
    for (int r = 0; r < nr; ++r) 
    {
      for (int c = 0; c < nc; ++c) 
      {
        if (grid[r][c] == '1') 
        {
          ++num_islands;
          grid[r][c] = '0'; // mark as visited，否则会重复探索，无限循环
          queue<pair<int, int>> neighbors;
          neighbors.push({r, c});
          while (!neighbors.empty())          //如果空，则说明 该 岛屿搜索完成
          {
            auto rc = neighbors.front();
            neighbors.pop();
            int row = rc.first, col = rc.second;        //队头元素为起点，向四周搜索
            if (row - 1 >= 0 && grid[row-1][col] == '1')            //上一行
            {														//满足判断条件则将位置（i,j）入队，待后续从队头pop出来进行BFS搜索
              neighbors.push({row-1, col}); grid[row-1][col] = '0';
            }
            if (row + 1 < nr && grid[row+1][col] == '1')            //下一行 
            {
              neighbors.push({row+1, col}); grid[row+1][col] = '0';
            }
            if (col - 1 >= 0 && grid[row][col-1] == '1')            //左一列 
            {
              neighbors.push({row, col-1}); grid[row][col-1] = '0';
            }
            if (col + 1 < nc && grid[row][col+1] == '1')            //右一列 
            {
            	neighbors.push({row, col+1}); grid[row][col+1] = '0';
            }
          }
        }
      }
    }
    return num_islands;
  }
};