有
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
解题思路
本文采用并查集思想:并查集是一种数据结构,它支持两种基本操作:
union
和find
。union
操作用于合并两个集合,而find
操作用于查找元素所属的集合。
本文中,如果两个城市相连则将他们通过union
方法**“连接”**起来,最后便可得知有几个省份
findC函数: 寻找某一个城市的根节点,一开始每个城市的根节点是自己,但如果两个城市连接后,根节点就会变成同一个
unionC函数: 将两个指定的城市的根节点设置为同一个,即达到“连接”的效果
class Solution {
public:
//寻找传入城市索引的根节点
int findC(vector<int>& parent,int index){
if(parent[index]!=index){
parent[index]=findC(parent,parent[index]);
}
return parent[index];
}
//通过将两个城市的根节点设置成同一个,以达到将两个城市连接的目的
void unionC(vector<int>& parent,int index1,int index2){
parent[findC(parent,index1)]=findC(parent,index2);
}
int findCircleNum(vector<vector<int>>& isConnected) {
int len=isConnected.size();
vector<int> parent(len);
//初始化每个城市的根节点都是自己
for(int i=0;i<len;i++){
parent[i]=i;
}
for(int i=0;i<len;i++){
//因为是从第一个城市开始遍历的且连接是双向的,所以后面的城市只需要往后遍历就行
for(int j=i+1;j<len;j++){
if(isConnected[i][j]==1){
unionC(parent,i,j);
}
}
}
//最终连接的城市都会是同一个根节点,所以只需要查找有几个根节点城市即可判断省份数量
int re=0;
for(int i=0;i<len;i++){
if(parent[i]==i){
++re;
}
}
return re;
}
};