LeetCode 200. 岛屿数量

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

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

示例 1:

输入:
11110
11010
11000
00000

输出: 1

示例 2:

输入:
11000
11000
00100
00011

输出: 3

解法

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

代码

class UnionFind {
public:
    vector<int> father;
    UnionFind(int num) {
        for (int i = 0; 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[4][2] = {{1,0},{-1,0},{0,1},{0,-1}};
    int encode(int i,int j,int n){
        return i*n+j;
    }
    int numIslands(vector<vector<char>>& grid) {
        int M = grid.size();
        if(M == 0)
            return 0;
        int N = grid[0].size();
        if(N == 0)
            return 0;
        UnionFind UF(M*N);
        for(int i = 0;i < M;i++){
            for(int j =0;j < N;j++){
                if(grid[i][j] == '1'){
                    for(int d =0;d < 4; d++){
                        int di = directions[d][0];
                        int dj = directions[d][1];
                        if(i+di >= 0 && i+di < M && j+dj >=0&& j+dj < N &&
                            grid[i+di][j+dj] == '1')
                        {
                            UF.Union(encode(i,j,N),encode(i+di,j+dj,N));
                        }
                    }
                }
            }
        }
        
        int res = 0;
        for(int i =0;i < M;i++){
            for(int j = 0; j< N;j++){
                if(grid[i][j] == '1'){
                    int id = encode(i,j,N);
                    if(UF.Find(id) == id)
                        res++;
                }
            }
        }
        
        return res;
        
    }
};