leetcode medium 树和图:岛屿数量

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

示例 1:

输入:
11110
11010
11000
00000

输出: 1

示例 2:

输入:
11000
11000
00100
00011

输出: 3

用队列来进行存储,首先第一个1进队,然后将上下左右的1加入,每一次加入队列都置0,然后出队一个,继续将上下左右的1加入,循环往复,直到队列为空

void EnQueue(vector<vector<char>>& grid, int x, int y,queue<pair<int,int>> &q) {
    // cout<<x<<" "<<y<<endl;
	if (x + 1 < grid.size() && grid[x + 1][y] == '1') q.push(make_pair(x + 1, y));
	if (y + 1 < grid[0].size() && grid[x][y + 1] == '1') q.push(make_pair(x, y + 1));
	if (x - 1 >= 0 && grid[x - 1][y] == '1') q.push(make_pair(x - 1, y));
	if (y - 1 >= 0 && grid[x][y - 1] == '1') q.push(make_pair(x, y - 1));
}
int numIslands(vector<vector<char>>& grid) {
	int res = 0;
	queue<pair<int, int>> q;
	for (int i = 0; i < grid.size(); i++) {
		for (int j = 0; j < grid[0].size(); j++) {
			if (grid[i][j] == '1') {
				res++;
				q.push(make_pair(i, j));
                // cout<<"2"<<endl;
				while (!q.empty()) {
                    // cout<<"1"<<endl;
					int x = q.front().first;
					int y = q.front().second;
					if(grid[x][y]=='1') EnQueue(grid, x, y, q);
                    grid[x][y] = 0;
                    q.pop();
				}
			}
		}
	}
	return res;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章