《并查集》200. 岛屿数量《leetcode》

200. 岛屿数量

难度中等858

给你一个由 '1'（陆地）和 '0'（水）组成的的二维网格，请你计算网格中岛屿的数量。

岛屿总是被水包围，并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。

此外，你可以假设该网格的四条边均被水包围。

 

示例 1：

输入：grid = [
  ["1","1","1","1","0"],
  ["1","1","0","1","0"],
  ["1","1","0","0","0"],
  ["0","0","0","0","0"]
]
输出：1

示例 2：

输入：grid = [
  ["1","1","0","0","0"],
  ["1","1","0","0","0"],
  ["0","0","1","0","0"],
  ["0","0","0","1","1"]
]
输出：3

 

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 300
• grid[i][j] 的值为 '0' 或 '1'

思路解析：

并查集： find ，union 

sum是图中元素为1的节点的个数。

对于图中元素为1的，更新其左边和上边的元素，构建子集，如果根不同，合并并sum-1。

需要注意的是，本次hashMap还是<Integer,Integer>,而不是class coordinate{int x,int y}  如果元素为类的话，会出现查询为空的情况。比如你再找的对应根时，会出现root.get(new coordinate（x,y）),这种情况会返回null，我认为是虽然你原来的类和现在新建的类的x和y相同，但是java在建立新类时会分配新的地址，从而和原来的类不同，因而为空。

AC代码：

class Solution {
    public int sum;
    public int dx[]={-1,0};
    public int dy[]={0,-1};
    Map<Integer,Integer> root=new HashMap<Integer,Integer>();
    
    public int getIndex(int i,int j,int n)
    {
        return i*n+j;
    }
    
    public boolean isOk(int x,int y,char grid[][])
    {
        return x>=0&&x<grid.length&&y>=0&&y<grid[0].length&&grid[x][y]=='1'? true:false;
    }
   
    public int find(int a)
    {
        Integer rootA=root.get(a);
        while(rootA!=root.get(rootA))
            rootA=root.get(root.get(rootA));//路径压缩
        return rootA;
    }
    public void union(int a,int b)
    {
        int rootA=find(a);
        int rootB=find(b);
        if(rootA!=rootB)
        {
            root.put(rootA,rootB);
            sum--;
        }
        return ;
    }

    public int numIslands(char[][] grid) {
        int m=grid.length;
        if(m==0)
            return 0;
        int n=grid[0].length;
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                if(isOk(i,j,grid))
                    {
                        root.put(getIndex(i,j,n),getIndex(i,j,n));
                        sum++;
                    }

        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                if(isOk(i,j,grid))//
                {       
                    if(isOk(i+dx[0],j+dy[0],grid))
                        union(getIndex(i,j,n),getIndex(i+dx[0],j+dy[0],n));
                    if(isOk(i+dx[1],j+dy[1],grid))
                        union(getIndex(i,j,n),getIndex(i+dx[1],j+dy[1],n));
                }        
            
        return sum;
    }
}

再提一点：

路径压缩有时候跑的比没有压缩的时间久。