leetcode 200. 岛屿数量(并查集)

	class findunion{
  public:
  vector<int>father;
  findunion(const int &num)
  {
    for(int i=0;i<num;i++)father.push_back(i);//让自己指向自己
     
  }
  int find(const int &n)
  {
    if(father[n]==n)return n;
    father[n]=find(father[n]);//路径压缩，让同集合指向一个元素
    
    return father[n];
  }
  
  void myunion(const int &a,const int &b)
  {
    if(father[a]==b||father[b]==a)return;//判断是否是同一个集合，如果不是就加进来
    int fa=find(a);
    int fb=find(b);
    father[fb]=fa;
  }
  //整一个是模板
  
};

class Solution {
public:
    int n,m;
    int encode(int &i,int &j,int &m)
    {
      return i*m+j;//二维转一维的集合
    }
    int numIslands(vector<vector<char>>& grid) {
       if(!grid.size()||!grid[0].size())return 0;
       n=grid.size(),m=grid[0].size();
       findunion F(n*m);
       int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
       
     
      for(int i=0;i<n;i++)
      for(int j=0;j<m;j++)
      {
        if(grid[i][j]=='1')
        {
          for(int k=0;k<4;k++)
          {
            int a=i+dx[k],b=j+dy[k];
            if(a>=0&&a<n&&b>=0&&b<m&&grid[a][b]=='1')
            F.myunion(encode(i,j,m),encode(a,b,m));//加入集合
          }
        }
      }
      int res=0;
      for(int i=0;i<n;i++)
      for(int j=0;j<m;j++)
      {
        if(grid[i][j]=='1'&&F.find(encode(i,j,m))==encode(i,j,m))//找到有几个集合老大
          res++;
      }
      return res;
    }
};