LeetCode#547 Friend Circles题解（C++版）

题干

原题网址：
https://leetcode.com/problems/friend-circles/description/

难度

中等（Medium）

知识点

图的深度优先搜索算法

题干解析

给你一个用二维数组表示的图，彼此有连接的点（直接或间接都可以）凑成一个“朋友圈”，要你算出这个图有多少个“朋友圈”。

解题思路

首先，用DFS（深度优先搜索算法）对图中的任意一个未被加入某朋友圈的点进行搜索，将此点先压入栈中，然后查找与此点相连接的点，一旦找到，将此点标记为已入朋友圈，并压入栈中，跳出循环。每次对栈顶的点进行搜索，一旦此点已经搜完其他所有的点方可弹掉。这样直到栈变成空一次，就是找到一个朋友圈。
接下来，继续用DFS搜索未被加入某朋友圈的点，直到所有的点都被标记为加入朋友圈，搜索结束，输出结果。

代码

class Solution {
public:
    int findCircleNum(vector<vector<int> >& M) {
        int n = M.size();
        int num = 0;
        set<int> used;
        while (used.size() < n) {
            stack<int> f;
            for (int i = 0; i < n; i++) {
                if (used.find(i) == used.end()) {
                    f.push(i);
                    used.insert(i);
                    break;
                }
            }
            while (!f.empty()) {
                int index = f.top();
                int i;
                for (i = 0; i < n; i++) {
                    if (used.find(i) == used.end() && M[index][i] == 1) {
                        f.push(i);
                        used.insert(i);
                        break;
                    }
                }
                if (i == n) {
                    f.pop();
                }
            }
            num++;
        }
        return num;
    }
};