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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章