547. Friend Circles 【Medium】 并查集

题目：给出N个学生之间是朋友的邻接矩阵，互为朋友的学生之间组成一个朋友圈，求出朋友圈的数量。

思路：就是要把N个学生划分到不同的集合中，集合中的人互相直接或间接是好朋友。

若用DFS的办法来做，划分学生的过程可能复杂度会比较高。这里采用了并查集的做法。

初始时，每个节点自成一个集合。

然后对N个节点的任意两个节点都要进行合并。每次合并两个节点x,y之前，先把x和y自身的先继和他们的先继节点的先继都设置到祖先节点（这样可以降低复杂度），然后再按照祖先的大小进行合并（取序号小的为根）。

最后，每个点的先继都设置成祖先，然后找出先继数组中不同祖先的个数就是朋友圈的个数了。

class Solution {
public:
	int find(vector<int>& root, int cur) {
		if (cur != root[cur]) root[cur] = find(root, root[cur]);
		return root[cur];
	}
    int findCircleNum(vector<vector<int> >& M) {
    	int n = M.size();
        vector<int> root(n, 0);
        for (int i = 0; i < n; ++i) root[i] = i;

        for (int i = 0; i < n - 1; ++i) {
        	for (int j = i + 1; j < n; ++j) {
        		if (M[i][j]) {
        			int ri = find(root, i), rj = find(root, j);
        			if (ri == rj)
        				continue;
        			else if (ri < rj) {
        				root[rj] = ri;
        			} else {
        				root[ri] = rj;
        			}
        		}
        	}
        }


	    int temp = -1, sum = 0;
	    for (int i = 0; i < n; ++i) {
        	root[i] = find(root, i);
	    	if (root[i] > temp) {
	    		temp = root[i];
	    		sum++;
	    	}
	    }
	    return sum;
    }
};