LeetCode 547. Friend Circles

本题题目大意是一个班有N个学生，其中一些彼此是朋友，如果朋友的朋友也是朋友，现在定义一个朋友圈包含一组彼此是朋友的学生，求一个班共有多少朋友圈。

题目的输入是一个邻接矩阵，很自然地可以想到用图的遍历算法来解题，可以用DFS从任意一点遍历一个朋友圈，若没有遍历完整个图，则从未被访问的一点开始继续遍历，重复这个过程直到遍历完整个图，这样就可以求出共有多少个朋友圈。

本题也是并查集的一个经典的场景，可以使用并查集，把有朋友关系的人都放到一个集合里，每个并查集都有一个代表，最后统计一共有多少个集合。初始化阶段，每个人都属于一个独立的集合，共有N个朋友圈，集合的代表就是他们自己。根据题意，合并的条件十分简单，如果两个学生是朋友，但是他们所属集合的代表不同，那就将两个集合合并，每合并一次朋友圈的数目就减少1，根据邻接矩阵合并完就能得到答案。

#include <iostream>#include <cstdlib>#include <vector>using namespace std;class Solution {    int pre[200];    int getParent(int stu) {        if (pre[stu] == stu)            return stu;        return getParent(pre[stu]);    }public:    int findCircleNum(vector<vector<int> >& M) {        int N = int(M.size());        for (int i = 0; i < N; ++i)            pre[i] = i;        int a, b;        int circleNum = N;        for (int i = 0; i < N; ++i)            for (int j = i + 1; j < N; ++j)                if (M[i][j]) {                    a = getParent(i);                    b = getParent(j);                    if (a != b) {                        pre[a] = b;                        circleNum--;                    }                }        return circleNum;    }};int main(int argc, char const *argv[]) {    Solution s;    vector<vector<int> > M;    int a[3][3] = { { 1, 1, 0 }, { 1, 1, 0 }, { 0, 0, 1 } };    for (int i = 0; i < 3; i++) {        vector<int> tmp;        for (int j = 0; j < 3; j++) {            tmp.push_back(a[i][j]);        }        M.push_back(tmp);    }    cout << s.findCircleNum(M) << endl;    system("pause");    return 0;}