200. 島嶼數量-並查集-中等
阿新 • • 發佈:2020-09-23
問題描述
給你一個由'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; } }