LintCode-岛屿的个数

433. 岛屿的个数

给一个01矩阵，求不同的岛屿的个数。

0代表海，1代表岛，如果两个1相邻，那么这两个1属于同一个岛。我们只考虑上下左右为相邻。

样例

在矩阵：

[
  [1, 1, 0, 0, 0],
  [0, 1, 0, 0, 1],
  [0, 0, 0, 1, 1],
  [0, 0, 0, 0, 0],
  [0, 0, 0, 0, 1]
]

中有 3 个岛.

标签

Zenefits 谷歌 脸书

相关题目

• 2 中等 大岛的数量 21 % 2 (breadth-first-search),(union-find) 中等 被围绕的区域 23 % 3 (union-find),(google) 困难 岛屿的个数II 19 %

DFS花了500多ms，人家都是几十ms搞定的，网上也没看到别的什么解法，用队列写了个BFS也没什么改善，很是怀疑评测机的性能。

---

DFS版：

class Solution {
public:
    /*
     * @param grid: a boolean 2D matrix
     * @return: an integer
     */
     vector<vector<bool>> grid;
     void DFS(const int& i,const int& j)
     {
         if(grid[i][j]==0) return;
         grid[i][j]=0;
         if(i>0) DFS(i-1,j);
         if(j>0) DFS(i,j-1);
         if(i<grid.size()-1) DFS(i+1,j);
         if(j<grid[0].size()-1) DFS(i,j+1);
     }
    int numIslands(vector<vector<bool>> &g) {
        // write your code here
        int count=0;
        grid=move(g);
        for(int i=0;i<grid.size();i++)
        {
            for(int j=0;j<grid[i].size();j++)
            {
                if(grid[i][j]!=0) 
                {
                    DFS(i,j);
                    count++;
                }
            }
        }
        return count;
    }
};

BFS版：

class Solution {
public:
    /*
     * @param grid: a boolean 2D matrix
     * @return: an integer
     */
     vector<vector<bool>> grid;
     queue<pair<int,int>> q;
     void BFS(const int& i,const int& j)
     {
         q.push(pair<int,int>(i,j));
         while(!q.empty())
         {
             int i=q.front().first;
             int j=q.front().second;
             grid[i][j]=0;
             q.pop();
             if(i>0 && grid[i-1][j]!=0) q.push(pair<int,int>(i-1,j));
             if(j>0 && grid[i][j-1]!=0) q.push(pair<int,int>(i,j-1));
             if(i<grid.size()-1 && grid[i+1][j]!=0) q.push(pair<int,int>(i+1,j));
             if(j<grid[0].size()-1 && grid[i][j+1]!=0) q.push(pair<int,int>(i,j+1));
         }
     }
    int numIslands(vector<vector<bool>> &g) {
        // write your code here
        int count=0;
        grid=move(g);
        for(int i=0;i<grid.size();i++)
        {
            for(int j=0;j<grid[i].size();j++)
            {
                if(grid[i][j]!=0) 
                {
                    BFS(i,j);
                    count++;
                }
            }
        }
        return count;
    }
};