程式碼隨想錄演算法訓練營 | 200.島嶼的數量(dfs/bfs)

W-Vicky11發表於2024-11-14

島嶼的數量

題目連結:https://leetcode.cn/problems/number-of-islands/

此題目要點:dfs和bfs都可以解決此題,但是使用dfs程式碼會更加的簡潔

  • 首先對grid進行遍歷,每一個節點都進行檢查,判斷是否是1(陸地)
  • 如果是,則進行dfs深搜,並將所有搜到的島嶼節點全置為0,表示此塊島嶼已經被搜過了,防止後序找根節點的時候又將搜過的島嶼計入。
  • grid遍歷的過程中如果再次遇到為1的節點,說明又有新的島嶼出現,數量+1
    image

dfs深搜

class Solution {
public:
    int ans=0;
    int m,n;
    int dir[4][2]={0,1,1,0,0,-1,-1,0};
    int numIslands(vector<vector<char>>& grid) {
        //找到grid的行數和列數
       m=grid.size();
       n=grid[0].size(); 
       //對grid進行遍歷
       for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(grid[i][j]=='1'){
                    dfs(grid,i,j);  //如果找到島嶼了,則進行dfs
                    ans++;
                }
            }
       }
       return ans;
    }
    void dfs(vector<vector<char>>& grid,int x,int y){
        grid[x][y]='0';
        for(int i=0;i<4;i++){
            int nx=x+dir[i][0];
            int ny=y+dir[i][1];
            if(nx>=0&&nx<m&&ny>=0&&ny<n&&grid[nx][ny]=='1') dfs(grid,nx,ny);
        }
    }
};

bfs廣搜

class Solution {
public:
    int ans=0;
    int m,n;
    queue<pair<int,int>> que;
    int dir[4][2]={0,1,1,0,0,-1,-1,0};
    int numIslands(vector<vector<char>>& grid) {
        //找到grid的行數和列數
       m=grid.size();
       n=grid[0].size(); 
       //對grid進行遍歷
       for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(grid[i][j]=='1'){
                    bfs(grid,i,j);  //如果找到島嶼了,則進行bfs
                    ans++;
                }
            }
       }
       return ans;
    }
    void bfs(vector<vector<char>>& grid,int x,int y){
        grid[x][y]='0';
        que.push({x,y});
        while(!que.empty()){
            auto cur=que.front(); que.pop();
            for(int i=0;i<4;i++){
            int nx=cur.first+dir[i][0];
            int ny=cur.second+dir[i][1];
            if(nx>=0&&nx<m&&ny>=0&&ny<n&&grid[nx][ny]=='1') bfs(grid,nx,ny);
        }
        }
    }
   
};

相關文章