给你一个由 '1'(陆地)和 '0'(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
示例 1:
输入:
11110
11010
11000
00000
输出: 1
示例 2:
输入:
11000
11000
00100
00011
输出: 3
解释: 每座岛屿只能由水平和/或竖直方向上相邻的陆地连接而成。
思路:
用广度优先算法遍历整张图,每次队列为空就说明遍历完了一个岛屿,岛屿数++。要注意不能重复访问相同的节点,需要用一个额外的数组来记录曾经访问过的点。(或者每次访问到‘1’时就将其改为‘0’)。还要注意判断输入是否为空,否则过不了leetcode 的测试点。
class Solution {
public:
int numIslands(vector<vector<char>>& grid) {
queue<pair<int,int> > q;
int n=(int)grid.size();
if (!n) return 0; //判断输入是否为空
int m=(int)grid[0].size();
vector<vector<bool> > visited(n,vector<bool>(m));
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
visited[i][j]=false;
}
}
int res=0;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(visited[i][j] || grid[i][j]=='0') continue;
q.push(pair<int,int>(i,j));
visited[i][j]=true;
while(!q.empty()){
pair<int,int> tmp=q.front();
int x=tmp.first;
int y=tmp.second;
q.pop();
if(x!=0){
if(grid[x-1][y]=='1' && !visited[x-1][y]){
q.push(pair<int, int>(x-1,y));
visited[x-1][y]=true;
}
}
if(x!=n-1){
if(grid[x+1][y]=='1' && !visited[x+1][y]){
q.push(pair<int, int>(x+1,y));
visited[x+1][y]=true;
}
}
if(y!=0){
if(grid[x][y-1]=='1' && !visited[x][y-1]){
q.push(pair<int, int>(x,y-1));
visited[x][y-1]=true;
}
}
if(y!=m-1){
if(grid[x][y+1]=='1' && !visited[x][y+1]){
q.push(pair<int, int>(x,y+1));
visited[x][y+1]=true;
}
}
}
res++;
}
}
return res;
}
};