岛屿数量的学习（BFS算法解答）

最新推荐文章于 2024-08-19 09:00:00 发布

sugarkss

最新推荐文章于 2024-08-19 09:00:00 发布

阅读量452

点赞数

分类专栏： C++学习

本文链接：https://blog.csdn.net/qq_41466376/article/details/106022816

版权

C++学习专栏收录该内容

8 篇文章 0 订阅

订阅专栏

学习基本算法——深度优先搜索（DFS）和广度优先搜索（BFS）
DFS和BFS

给你一个由 ‘1’（陆地）和 ‘0’（水）组成的的二维网格，请你计算网格中岛屿的数量。

岛屿总是被水包围，并且每座岛屿只能由水平方向或竖直方向上相邻的陆地连接形成。

此外，你可以假设该网格的四条边均被水包围。

示例 1:

输入:
11110
11010
11000
00000
输出: 1
示例 2:

输入:
11000
11000
00100
00011
输出: 3
解释: 每座岛屿只能由水平和/或竖直方向上相邻的陆地连接而成。

解题思路

根据采用BFS算法，即广度优先搜索算法（宽度）。我们可以从原点（0,0）开始从上往下，从左往右的顺序遍历整个二维数组，例如当（0,0）值为1，则将其加入处理队列，（若值为‘0’则队列是空，继续往下遍历，即（0，1）位置），然后判断处理队列是否为空，不为空，将其周围的点进行判断，判断是陆地（‘1’）还是海洋（‘0’），是陆地点，则将其加入处理队列，同时将其值置为‘0’。判断好该点四周的点是陆地还是海洋，（此时是在while循环中），等到该队列处理完所有点后，队列为空时，岛屿数量count++；就这样如此往复遍历整个二维数组。

#include<iostream>
#include<vector>
#include<queue>
#include<string>
using namespace std;
class Solution {
private:
	queue<pair<int, int>> que;//处理队列
	int x = 0;//处理队列中点的x
	int y = 0;//处理队列中点的y
	int fx = 0;//处理点四周的点的x
	int fy = 0;//处理点四周的点的x
	int count = 0;//岛屿数量

public:
	int numIslands(vector<vector<char>>& grid) {
		int rows = grid.size();
		int lines = grid[0].size();
		//out << rows << " " << lines << endl;
		//dx、dx四周点
		int dx[] = { -1,0,1,0 };
		int dy[] = { 0,1,0,-1 };

		if (rows == 0 || lines == 0) {
			return 0;
		}

		for (int i = 0; i < rows; i++) {
			for (int j = 0; j < lines; j++) {
				//第一次搜索到岛屿‘1’，将其加入队列进行处理，同时将其值置为‘0’，也就说将该点变为海洋，方便之后的处理
				if (grid[i][j] == '1') {
					que.push(make_pair(i,j));
					grid[i][j] = '0';
					
					//判断处理队列是否为空
					while (!que.empty()) 
					{
						pair<int, int> node;
						node = que.front();
						x = node.first;
						y = node.second;
						que.pop();
						//循环四次，就是遍历了该点上下左右四个点
						for (int k = 0; k < 4; k++) {
							fx = x + dx[k];
							fy = y + dy[k];
							if (fx < 0 || fx >= rows || fy < 0 || fy >= lines) { continue; }
							//处理点周围点是否是海洋1还是陆地，是陆地说明该附近点与该点相连，将其加入处理队列，重复上面的操作。
							if (grid[fx][fy] == '1') {
								grid[fx][fy] = '0';
								que.push(make_pair(fx,fy));
							}
						}

					}
					count++;
				}
			}
		}
		return count;
	}
};



int main() {
	Solution s;
	vector<vector<char>> setchar{
		{'1', '1', '0', '0', '0' },
		{'1', '1', '0', '0', '0'},
	   { '0', '0', '1', '0', '0' },
		{ '0', '0', '0', '1', '1' } };

	cout << "A岛屿的数量：" << s.numIslands(setchar) << endl;


	Solution s2;
	vector<vector<char>> setchar2{
		{'1', '1', '1', '1', '0' },
		{'1', '1', '0', '1', '0'},
	   { '1', '1', '0', '0', '0' },
		{ '0', '0', '0', '0', '0' } };

	cout << "B岛屿的数量：" << s2.numIslands(setchar2) << endl;


	Solution s3;
	vector<vector<char>> setchar3{
		{'1', '0', '1', '1', '0' },
		{'0', '1', '0', '0', '1'},
	   { '0', '1', '0', '0', '1' },
		{ '0', '0', '1', '1', '0' } };

	cout << "C岛屿的数量：" << s3.numIslands(setchar3) << endl;

	return 0;
}