Leetcode 547. Friend Circles

SnailTyan

于 2018-08-08 22:27:15 发布

阅读量302

点赞数

分类专栏： Leetcode 文章标签： Leetcode

本文链接：https://blog.youkuaiyun.com/Quincuntial/article/details/81516740

版权

Leetcode 同时被 2 个专栏收录

409 篇文章

订阅专栏

Algorithm

260 篇文章

订阅专栏

文章作者：Tyan
博客：noahsnail.com | 优快云 | 简书

1. Description

Friend Circles

2. Solution

class Solution {
public:
    int findCircleNum(vector<vector<int>>& M) {
        int rows = M.size();
        int flag[rows] = {0};
        int count = 0;
        for(int i = 0; i < rows; i++) {
            if(!flag[i]) {
                ++count;
                queue<int> friends;
                friends.push(i);
                oneCircle(M, flag, friends);
            }
        }
        return count;
    }

    void oneCircle(vector<vector<int>>& M, int flag[], queue<int> friends) {
        int rows = M.size();
        while(!friends.empty()) {
            int row = friends.front();
            friends.pop();
            for(int i = 0; i < rows; i++) {
                if(M[row][i] && i != row && !flag[i]) {
                    flag[i] = 1;
                    friends.push(i);
                }
            }
        }
    }
};