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;
}