有 n 个城市,其中一些彼此相连,另一些没有相连。如果城市 a 与城市 b 直接相连,且城市 b 与城市 c 直接相连,那么城市 a 与城市 c 间接相连。
省份 是一组直接或间接相连的城市,组内不含其他没有相连的城市。
给你一个 n x n 的矩阵 isConnected ,其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连,而 isConnected[i][j] = 0 表示二者不直接相连。
返回矩阵中 省份 的数量。
简要题解
显然可以看出这是一个并查集的模板题 直接把相互连通的放在一个组内即可。
并查集部分主要代码如下 首先初始化代码假设为一个森林主要代码如下
void init()
{
for(int i=0;i<n;i++)fa[i]=i;
}
找祖先代码主要如下
int find(int x)
{
return fa[x]==x?x:(fa[x]=find(fa[x]));
}
合并的代码如下
void union(int x,int y)
{
fa[find(x)]=find(y);
}
该题主要代码如下所示
class Solution {
public:
vector<int>fa;
int find(int x)
{
if(fa[x]!=x) fa[x]=find(fa[x]);
return fa[x];
}
int findCircleNum(vector<vector<int>>& isConnected) {
int n=isConnected.size();
for(int i=0;i<n;i++)fa.push_back(i);
int ans=n;
for(int i=0;i<n;i++)
for(int j=0;j<i;j++)
{
if(isConnected[i][j]==1&&find(i)!=find(j))
{
fa[find(i)]=find(j);
ans--;
}
}
return ans;
}
};