LeetCode 200. 岛屿数量

习题地址 https://leetcode-cn.com/problems/number-of-islands/

给定一个由 '1'（陆地）和 '0'（水）组成的的二维网格，计算岛屿的数量。一个岛被水包围，并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。

示例 1:

输入:

输出: 

示例 2:

输入:

输出: 

解法

广度遍历或者使用并查集 这里使用并查集。注意将二维数组转换为一维的方法

代码

 class UnionFind {
 public:
     vector<int> father;
     UnionFind(int num) {
         for (int i = ; i < num; i++) {
             father.push_back(i);    //每个人都指向自己
         }
     }
     int Find(int n) {
         //非递归版本
         /*
         while (father[n] != n) {
             n = father[n];
         }
         return n;
         */
         //递归
         if (father[n] == n)
             return n;
         father[n] = Find(father[n]);
         return father[n] ;
     }
     void Union(int a, int b) {
         int fa = Find(a);
         int fb = Find(b);
         father[fb] = fa;
     }
 };

 class Solution {
 public:
     int directions[][] = {{,},{-,},{,},{,-}};
     int encode(int i,int j,int n){
         return i*n+j;
     }
     int numIslands(vector<vector<char>>& grid) {
         int M = grid.size();
         if(M == )
             return ;
         int N = grid[].size();
         if(N == )
             return ;
         UnionFind UF(M*N);
         for(int i = ;i < M;i++){
             for(int j =;j < N;j++){
                 if(grid[i][j] == ''){
                     for(int d =;d < ; d++){
                         int di = directions[d][];
                         int dj = directions[d][];
                         if(i+di >=  && i+di < M && j+dj >=&& j+dj < N &&
                             grid[i+di][j+dj] == '')
                         {
                             UF.Union(encode(i,j,N),encode(i+di,j+dj,N));
                         }
                     }
                 }
             }
         }

         int res = ;
         for(int i =;i < M;i++){
             for(int j = ; j< N;j++){
                 if(grid[i][j] == ''){
                     int id = encode(i,j,N);
                     if(UF.Find(id) == id)
                         res++;
                 }
             }
         }

         return res;

     }
 };