这次学习了代码随想录 岛屿数量-深度搜索 岛屿数量-广度搜索
本题思路,是用遇到一个没有遍历过的节点陆地,计数器就加一,然后把该节点陆地所能遍历到的陆地都标记上。在遇到标记过的陆地节点和海洋节点的时候直接跳过。 这样计数器就是最终岛屿的数量。那么在搜索遍历时,使用BFS,DFS都是可行的。
对于每一个节点,搜索时,需要向其上下左右四个反向进行搜索,使用 int dir[4][2] = {1, 0, 0, 1, -1, 0, 0, -1}; 来记录上下左右四个方向,使用一个 vector<vector<bool>> 来记录已经遍历过的节点,搜索时遇到已经遍历的节点,或者越界的节点就停止。
- 在dfs时,要注意 nextx,nexty的更新,for(int i = 0; i < 4; i++) 每次都会向一个方向进行移动,都是在原本的x,y基础上向一个方向移动,因此不能写成 x += dir[i][0]; y += dir[i][1]; 这种形式
- 在bfs时,要注意nextx,nexty的更新,for(int i = 0; i < 4; i++)每次移动的对象都是新出队的节点,而不是bfs传入的x,y;每次while循环都会有新的节点出队,是这个新出队的节点在移动。同时要注意标记遍历 的时刻,一个节点入队后,就需要将其状态修改为 true,防止上一层的节点重复遍历该节点,让该节点重复入队。
#include<bits/stdc++.h>
using namespace std;
int dir[4][2] = {1, 0, 0, 1, -1, 0, 0, -1};
void dfs(vector<vector<int>>& graph, vector<vector<bool>>& visited, int x, int y){
for(int i = 0; i < 4; i++){
int nextx = x + dir[i][0];
int nexty = y + dir[i][1];
if(nextx < 0 || nextx >= graph.size() || nexty < 0 || nexty >= graph[0].size()){
continue;
}else{
if(graph[nextx][nexty] == 1 && !visited[nextx][nexty]){
visited[nextx][nexty] = true;
dfs(graph, visited, nextx, nexty);
}
}
}
}
void bfs(vector<vector<int>>& graph, vector<vector<bool>>& visited, int x, int y){
queue<pair<int, int>> que;
que.push({x, y});
while(!que.empty()){
pair<int, int> cur = que.front();
que.pop();
for(int i = 0; i < 4; i++){
int nextx = cur.first + dir[i][0];
int nexty = cur.second + dir[i][1];
if(nextx < 0 || nextx >= graph.size() || nexty < 0 || nexty >= graph[0].size()){
continue;
}else{
if(graph[nextx][nexty] == 1 && !visited[nextx][nexty]){
que.push({nextx, nexty});
visited[nextx][nexty] = true;
}
}
}
}
}
int main(){
int n,m;
cin >> n >> m;
vector<vector<int>> graph(n, vector<int>(m, 0));
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
cin >> graph[i][j];
}
}
vector<vector<bool>> visited(n, vector<bool>(m, false));
int result = 0;
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
if(graph[i][j] == 1 && !visited[i][j]){
result++;
visited[i][j] = true;
//dfs(graph, visited, i, j);
bfs(graph, visited, i, j);
}
}
}
cout << result << endl;
return 0;
}