Leetcode 200.岛屿的数量 - DFS、BFS
Leetcode 200 岛屿的数量:
DFS利用函数调用栈保证了检索顺序,
BFS则需要自己建立队列,把待检索对象按规则入队。
class Solution { // DFS解法,8ms/10.7MB,99.7% / 92%
public:
/**
* row,col: 坐标,以0开始.
* 用row和col,而不是x,y. 否则容易写成grid[x][y],顺序不对!!
*/
void dfs_set_zero(vector<vector<char>>& grid, int row, int col) {
int n_rows = grid.size();
if (!n_rows) return;
int n_cols = grid[0].size();
if (!n_cols) return;
if (row<0 || row>=n_rows || col<0 || col>=n_cols) return;
if (grid[row][col] == '1') {
grid[row][col] = '0';
dfs_set_zero(grid, row-1, col);
dfs_set_zero(grid, row+1, col);
dfs_set_zero(grid, row, col-1);
dfs_set_zero(grid, row, col+1);
}
}
int numIslands(vector<vector<char>>& grid) {
int result = 0;
for (int i=0; i<grid.size(); ++i) {
for (int j=0; j<grid[0].size(); ++j) {
if (grid[i][j] == '1') {
result++;
dfs_set_zero(grid, i, j);
}
}
}
return result;
}
};
class Solution { // BFS解法,12ms/11MB, 96.9% / 69.4%
public:
queue<pair<int,int>> q;
void bfs_set_zero(vector<vector<char>>& grid) {
int n_rows = grid.size();
if(!n_rows) return;
int n_cols = grid[0].size();
if(!n_cols) return;
while (!q.empty()) {
auto pos = q.front();
q.pop();
//if (grid[pos.first][pos.second] == '1') {
grid[pos.first][pos.second] = '0';
if (pos.first-1>=0 && grid[pos.first-1][pos.second] == '1')
q.emplace(pos.first-1,pos.second);
if (pos.first+1<n_rows && grid[pos.first+1][pos.second] == '1')
q.emplace(pos.first+1,pos.second);
if (pos.second-1>=0 && grid[pos.first][pos.second-1] == '1')
q.emplace(pos.first,pos.second-1);
if (pos.second+1<n_cols && grid[pos.first][pos.second+1] == '1')
q.emplace(pos.first,pos.second+1);
//}
}
}
int numIslands(vector<vector<char>>& grid) {
int result = 0;
for (int i=0; i<grid.size(); i++) {
for (int j=0; j<grid[0].size(); j++) {
if (grid[i][j] == '1') {
result++;
//q.push(make_pair(i,j));
q.emplace(i,j); // 使用emplace而非push,constructed in-place
// 其实这种方式和 q.push( {i,j} );是相同的(右值引用),详见 “C++11的初始化方式”https://blog.csdn.net/K346K346/article/details/55194246
bfs_set_zero(grid);
}
}
}
return result;
}
};