1. 程式人生 > 實用技巧 >200. 島嶼數量-並查集-中等

200. 島嶼數量-並查集-中等

問題描述

給你一個由'1'(陸地)和 '0'(水)組成的的二維網格,請你計算網格中島嶼的數量。

島嶼總是被水包圍,並且每座島嶼只能由水平方向或豎直方向上相鄰的陸地連線形成。

此外,你可以假設該網格的四條邊均被水包圍。

示例 1:

輸入:
[
['1','1','1','1','0'],
['1','1','0','1','0'],
['1','1','0','0','0'],
['0','0','0','0','0']
]
輸出:1
示例2:

輸入:
[
['1','1','0','0','0'],
['1','1','0','0','0'],
['0','0','1','0','0'],
['0','0','0','1','1']
]
輸出: 3
解釋: 每座島嶼只能由水平和/或豎直方向上相鄰的陸地連線而成。

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/number-of-islands

解答

//並查集中連通分量的數目
class Solution {
    int[] parents;
    public int find(int x){
        if(parents[x] == -1)return x;
        return find(parents[x]);
    }
    public void union(int x, int y){
        int xP = find(x);
        int yP = find(y);
        
if(xP != yP)parents[xP] = yP; } public int numIslands(char[][] grid) { int r = grid.length; if(r == 0)return 0; int c = grid[0].length; if(c == 0)return 0; int len = r*c; parents = new int[r*c]; Arrays.fill(parents, -2); for(int i=0;i<r;i++)
for(int j=0;j<c;j++) if(grid[i][j] == '1')parents[i*c+j] = -1; for(int i=0;i<r;i++) for(int j=0;j<c;j++){ if(grid[i][j] == '1'){ if(i<r-1 && grid[i+1][j] == '1')union((i+1)*c+j, i*c+j); if(j<c-1 && grid[i][j+1] == '1')union(i*c+j, i*c+j+1); } } int islands=0; for(int i:parents) if(i==-1)islands++; return islands; } }