547. 省份数量
一、题目
有 n
个城市,其中一些彼此相连,另一些没有相连。如果城市 a
与城市 b
直接相连,且城市 b
与城市 c
直接相连,那么城市 a
与城市 c
间接相连。
省份 是一组直接或间接相连的城市,组内不含其他没有相连的城市。
给你一个 n x n
的矩阵 isConnected
,其中 isConnected[i][j] = 1
表示第 i
个城市和第 j
个城市直接相连,而 isConnected[i][j] = 0
表示二者不直接相连。
返回矩阵中 省份 的数量。
二、思路
遍历矩阵 isConnected 的全部元素之后,计算连通分量的总数,即为省份的总数。
三、代码
class Solution { public int findCircleNum(int[][] isConnected) { int cities = isConnected.length; int[] parent = new int[cities]; for (int i = 0; i < cities; i++) { parent[i] = i; } for (int i = 0; i < cities; i++) { for (int j = i + 1; j < cities; j++) { if (isConnected[i][j] == 1) { union(parent, i, j); } } } int provinces = 0; for (int i = 0; i < cities; i++) { if (parent[i] == i) { provinces++; } } return provinces; } public void union(int[] parent, int index1, int index2) { parent[find(parent, index1)] = find(parent, index2); } public int find(int[] parent, int index) { if (parent[index] != index) { parent[index] = find(parent, parent[index]); } return parent[index]; } }