并查集 LeetCode 547

LeetCode 547. 朋友圈

班上有 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。

示例 2:

输入: 
[[1,1,0],
 [1,1,1],
 [0,1,1]]
输出: 1
说明：已知学生0和学生1互为朋友，学生1和学生2互为朋友，所以学生0和学生2也是朋友，所以他们三个在一个朋友圈，返回1。

注意：

1. N 在[1,200]的范围内。
2. 对于所有学生，有M[i][i] = 1。
3. 如果有M[i][j] = 1，则有M[j][i] = 1.

思路

使用并查集这个数据结构，将是朋友的两个同学所在集合合并，最终输出集合的总数即可

并查集数据结构设计参考另外一篇博客：https://blog.csdn.net/qq_31965925/article/details/106265355

 

代码

//思路：使用并查集 
class Solution {
private:
	vector<int> parent;  
	//查找节点p的根节点，并且在查找过程中使用路径压缩优化树结构 
	int find(int p)
	{
		if(p!=parent[p])
			parent[p] = find(parent[p])	;
		return parent[p];
	} 
public:
    int findCircleNum(vector<vector<int> >& M) {
		int n = M.size();
		if(n==0) return 0;
		//初始化parent
		for(int i=0; i<n; i++)
			parent.push_back(i);//当前节点根节点指向自己
		
 		for(int i=0; i<n; i++)
 		{
 			for(int j=0; j<n; j++)
 			{
 				// 学生i,学生j是朋友 合并他们的朋友圈
                if(M[i][j]==1)
                {
                    int pRoot = find(i);
				    int qRoot = find(j);
				    if(pRoot!=qRoot) 
					    parent[pRoot] = qRoot; 
                }
				
 			}
 		}
 		//查看当前的集合数量，就是查找不同的根节点有几个
		set<int> s;
		for(int i=0; i<n; i++)
		{
			s.insert(find(parent[i]));
		} 
		return s.size();	
    }
};