200. Number of Islands [union find]

https://leetcode-cn.com/problems/number-of-islands/

[bug]

class Solution {

    char[][] grid;
    int w, h;
    int[][] dirs = {{-1,0}, {1,0}, {0,-1}, {0,1}};
    Map<Integer, Integer> map = new HashMap<>();

    public int numIslands(char[][] grid) {

        this.grid = grid;
        int w = grid[0].length; 
        int h = grid.length;

        unionFind();
                     
        Set<Integer> rep = new HashSet<>();                      
        for(int y=0;y<h;y++){
            for(int x=0;x<w;x++){
                if(grid[y][x]=='1'){
                    int[] root = findRoot(y, x);
                    rep.add(toNum(root[0], root[1]));
                }
            }
        }
        return rep.size();
    }

    void unionFind(){

        for(int y=0;y<h;y++){
            for(int x=0;x<w;x++){
                if(grid[y][x]=='0'){
                    continue;
                }
                map.put(y*w+x, y*w+x);
                for(int[] dir:dirs){
                    int y2 = y + dir[0];
                    int x2 = x + dir[1];
                    if(bound(y2, x2)){
                        continue;
                    }
                    if(grid[y2][x2]=='0'){
                        continue;
                    }
                    int[] root1 = findRoot(y, x);   
                    int[] root2 = findRoot(y2, x2);
                    int rep1 = toNum(root2[0], root2[1]);
                    int rep2 = toNum(root1[0], root1[1]);
                    if(rep1==rep2){
                        continue;
                    }
                    map.put(rep2, rep1);
                }
            }
        }
        
    }

    int toNum(int y, int x){
        return y*w+x;
    }

    boolean bound(int y, int x){
        return y<0 || y>=h || x<0 || x>w;
    }

    int[] findRoot(int y, int x){
        int k = y*w + x;
        while(map.get(k)!=k){
            int v = map.get(k);
            k = v; 
        }
        int y2 = k/w;
        int x2 = k%w;
        return new int[]{y, x};
    }
}

[bug]

class Solution {

    char[][] grid;
    int w, h;
    int[][] dirs = {{-1,0}, {1,0}, {0,-1}, {0,1}};
    Map<Integer, Integer> map = new HashMap<>();

    public int numIslands(char[][] grid) {

        this.grid = grid;
        int w = grid[0].length; 
        int h = grid.length;

        unionFind();
                     
        Set<Integer> rep = new HashSet<>();                      
        for(int y=0;y<h;y++){
            for(int x=0;x<w;x++){
                if(grid[y][x]==1){
                    int[] root = findRoot(y, x);
                    rep.add(toNum(root[0], root[1]));
                }
            }
        }
        return rep.size();
    }

    void unionFind(){

        for(int y=0;y<h;y++){
            for(int x=0;x<w;x++){
                if(grid[y][x]==0){
                    continue;
                }
                map.put(y*w+x, y*w+x);
                for(int[] dir:dirs){
                    int y2 = y + dir[0];
                    int x2 = x + dir[1];
                    if(grid[y2][x2]==0){
                        continue;
                    }
                    int[] root1 = findRoot(y, x);   
                    int[] root2 = findRoot(y2, x2);
                    int rep1 = toNum(root2[0], root2[1]);
                    int rep2 = toNum(root1[0], root1[1]);
                    if(rep1==rep2){
                        continue;
                    }
                    map.put(rep2, rep1);
                }
            }
        }
        
    }

    int toNum(int y, int x){
        return y*w+x;
    }

    int[] findRoot(int y, int x){
        int k = y*w + x;
        while(map.get(k)!=k){
            int v = map.get(k);
            k = v; 
        }
        int y2 = k/w;
        int x2 = k%w;
        return new int[]{y, x};
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值