547. 朋友圈
班上有 N 名学生。其中有些人是朋友,有些则不是。他们的友谊具有是传递性。如果已知 A 是 B 的朋友,B 是 C 的朋友,那么我们可以认为 A 也是 C 的朋友。所谓的朋友圈,是指所有朋友的集合。
给定一个 N * N 的矩阵 M,表示班级中学生之间的朋友关系。如果M[i][j] = 1,表示已知第 i 个和 j 个学生互为朋友关系,否则为不知道。你必须输出所有学生中的已知的朋友圈总数。
解题思路: 本题本质上是在求图的连通域个数的问题,基本想到的解题方法主要有,DFS/BFS,从标签上看,这里也可以用并查集解题。
并查集:初始化一个数组,root[i]=i,每个元素分为一个单独的类,然后设计两个函数Find和Union。
class Solution {
public:
int findCircleNum(vector<vector<int>>& M) {
int n = M.size();
int res = n;
root.resize(n);
for (int i = 0; i < n; ++i) {
root[i] = i;
}
for (int i = 0; i < n; ++i) {
for (int j = i + 1; j < n; ++j) {
if (M[i][j] == 1) {
int r1 = Find(i);
int r2 = Find(j);
if (r1 != r2) {
--res;
root[r2] = r1;
}
}
}
}
return res;
}
private:
int Find(int i) {
while (i != root[i]) {
i = root[i];
}
return i;
}
vector<int> root;
};
参考资料
https://www.cnblogs.com/grandyang/p/6686983.html