难度中等
有 n
个城市,其中一些彼此相连,另一些没有相连。如果城市 a
与城市 b
直接相连,且城市 b
与城市 c
直接相连,那么城市 a
与城市 c
间接相连。
省份 是一组直接或间接相连的城市,组内不含其他没有相连的城市。
给你一个 n x n
的矩阵 isConnected
,其中 isConnected[i][j] = 1
表示第 i
个城市和第 j
个城市直接相连,而 isConnected[i][j] = 0
表示二者不直接相连。
返回矩阵中 省份 的数量。
示例 1:
输入:isConnected = [[1,1,0],[1,1,0],[0,0,1]] 输出:2
示例 2:
输入:isConnected = [[1,0,0],[0,1,0],[0,0,1]] 输出:3
提示:
1 <= n <= 200
n == isConnected.length
n == isConnected[i].length
isConnected[i][j]
为1
或0
isConnected[i][i] == 1
isConnected[i][j] == isConnected[j][i]
public int findCircleNum(int[][] isConnected) {
int[] parent = new int[isConnected.length];
for (int i = 0; i < parent.length; i++) {
parent[i] = i;
}
for (int i = 0; i < isConnected.length; i++) {
for (int j = 1; j < isConnected[0].length; j++) {
if (isConnected[i][j] == 1){
union(i,j,parent);
}
}
}
int count = 0;
for (int i = 0; i < parent.length; i++) {
if (parent[i] == i){
count++;
}
}
return count;
}
private void union(int i, int j, int[] parent) {
parent[find(i,parent)] = find(j,parent); //这个只是串联到一起,J 比 I 可能小
}
private int find(int i, int[] parent) {
if (parent[i] != i){
parent[i] = find(parent[i],parent); //这个是递归一定能找到最父的
}
return parent[i];
}