Leetcode笔记目录
一、题目描述
班上有 N 名学生。其中有些人是朋友,有些则不是。他们的友谊具有是传递性。如果已知 A 是 B 的朋友,B 是 C 的朋友,那么我们可以认为 A 也是 C 的朋友。所谓的朋友圈,是指所有朋友的集合。
给定一个 N * N 的矩阵 M,表示班级中学生之间的朋友关系。如果M[i][j] = 1,表示已知第 i 个和 j 个学生互为朋友关系,否则为不知道。你必须输出所有学生中的已知的朋友圈总数。
示例1:
- 输入:
[[1,1,0],
[1,1,0],
[0,0,1]]
输出: 2
解释: 已知学生 0 和学生 1 互为朋友,他们在一个朋友圈。第2个学生自己在一个朋友圈。所以返回 2 。
二、解题过程
1.思想
本题是一个求解无向图中连通分量个数的问题,可以用dfs,bfs,并查集,这里采用两种递归实现dfs和并查集;
2.代码
不判断的dfs递归实现:
int dfs(vector<vector<int>>& M,vector<bool>& v,int k){
if(v[k]==true)return 0;
v[k]=true;
for(int i=0;i<M.size();i++){
if(M[k][i]==1)
dfs(M,v,i);
}
return 1;
}
int findCircleNum(vector<vector<int>>& M) {
int num=0;
vector<bool> visit(M.size(),false);
for(int i=0;i<M.size();i++){
num += dfs(M,visit,i);
}
return num;
}
};
- 注意
- 此时不像Leetcode695可以数组值代替判断是否访问过,新增一个辅助数组visit是图遍历算法的常规操作;
先判断的dfs递归实现:
void dfs(vector<vector<int>>& M,vector<bool>& v,int k){
v[k]=true;
for(int i=0;i<M.size();i++){
if(M[k][i]==1&&v[k]!=true)
dfs(M,v,i);
}
}
int findCircleNum(vector<vector<int>>& M) {
int num=0;
vector<bool> visit(M.size(),false);
for(int i=0;i<M.size();i++){
if(!visit[i]){
dfs(M,visit,i);
num++;
}
}
return num;
}
};
- 注意
- 先判断的结构;
并查集实现:
int findFather(vector<int>& father,int k){
int a=k;
while(k!=father[k])k=father[k];
while(a!=father[a]){
int z=a;
a=father[a];
father[z]=k;
}
return k;
}
void Union(vector<int>& father,int k,int j){
int a=findFather(father,k);
int b=findFather(father,j);
if(a!=b)father[a]=b;
}
int findCircleNum(vector<vector<int>>& M) {
vector<int> father(M.size()); //错误2
int num=0;
for(int i=0;i<M.size();i++) father[i]=i;
for(int i=0;i<M.size();i++)
for(int j=i+1;j<M[0].size();j++){
if(M[i][j]==1)Union(father,i,j);
}
for(int i=0;i<M.size();i++)
if(father[i]==i)num++;
return num;
}
- 注意
- 此处并查集使用了压缩路径,但没有使用其他的比如平衡策略等加快并查集的方法,速度仍然可观;
- vector若使用push_back赋值,则声明时不要加范围,若使用一般数组赋值,则声明时一定要加范围;
三、总结
标准的无向图中连通分量个数的问题。