首页 > 其他 > 详细

Union Find-547. Friend Circles

时间:2018-01-13 10:34:34      阅读:76      评论:0      收藏:0      [点我收藏+]

标签:mon   relation   turn   ans   other   nat   rec   them   number   

There are N students in a class. Some of them are friends, while some are not. Their friendship is transitive in nature. For example, if A is a direct friend of B, and B is a direct friend of C, then A is an indirect friend of C. And we defined a friend circle is a group of students who are direct or indirect friends.

Given a N*N matrix M representing the friend relationship between students in the class. If M[i][j] = 1, then the ith and jth students are direct friends with each other, otherwise not. And you have to output the total number of friend circles among all the students.

Example 1:

Input: 
[[1,1,0],
 [1,1,0],
 [0,0,1]]
Output: 2
Explanation:The 0th and 1st students are direct friends, so they are in a friend circle. 
The 2nd student himself is in a friend circle. So return 2.

 

Example 2:

Input: 
[[1,1,0],
 [1,1,1],
 [0,1,1]]
Output: 1
Explanation:The 0th and 1st students are direct friends, the 1st and 2nd students are direct friends, 
so the 0th and 2nd students are indirect friends. All of them are in the same friend circle, so return 1.

 

Note:

  1. N is in range [1,200].
  2. M[i][i] = 1 for all students.
  3. If M[i][j] = 1, then M[j][i] = 1.
class Solution {
public:
    int dfs(vector<vector<int>>& M, bool *visited, int ind){
        if(visited[ind] == 1) return 0;
        else{
            visited[ind] = 1;
            int ccnum = 1;
            for(int j = 0; j < M.size(); j++){
                if(ind != j && M[ind][j] == 1){
                    ccnum += dfs(M, visited, j);
                }
            }
            return ccnum;
        }

    }
    int findCircleNum(vector<vector<int>>& M) {
        int size = M.size();
        if(size == 0) return 0;
        else{
            bool visited[size];
            memset(visited, 0, size);
            int cnum = 0;
            for(int i = 0; i < size; i++){
                if(dfs(M, visited, i) > 0)
                cnum++;
            }
            return cnum;
        }

    }
};

 

Union Find-547. Friend Circles

标签:mon   relation   turn   ans   other   nat   rec   them   number   

原文:https://www.cnblogs.com/msymm/p/8278247.html

(0)
(0)
   
举报
评论 一句话评论(0
0条  
登录后才能评论!
© 2014 bubuko.com 版权所有 鲁ICP备09046678号-4
打开技术之扣,分享程序人生!
             

鲁公网安备 37021202000002号