朋友圈

flag

软件学院大三党，每日一道算法题，第九天

题目介绍

班上有 N 名学生。其中有些人是朋友，有些则不是。他们的友谊具有是传递性。如果已知 A 是 B 的朋友，B 是 C 的朋友，那么我们可以认为 A 也是 C 的朋友。所谓的朋友圈，是指所有朋友的集合。

给定一个 N * N 的矩阵 M，表示班级中学生之间的朋友关系。如果M[i][j] = 1，表示已知第 i 个和 j 个学生互为朋友关系，否则为不知道。你必须输出所有学生中的已知的朋友圈总数。

示例 1:

输入:
[[1,1,0],
[1,1,0],
[0,0,1]]
输出: 2
说明：已知学生0和学生1互为朋友，他们在一个朋友圈。
第2个学生自己在一个朋友圈。所以返回2。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/friend-circles

思路

给定的矩阵可以看成图的邻接矩阵。这样我们的问题可以变成无向图连通块的个数。

①为了找到连通块的个数，一个简单的方法就是使用深度优先搜索，从每个节点开始，我们使用一个大小为 N 的 visited 数组（M 大小为 N×N），这样 visited[i] 表示第 i 个元素是否被深度优先搜索访问过。

我们首先选择一个节点，访问任一相邻的节点。然后再访问这一节点的任一相邻节点。这样不断遍历到没有未访问的相邻节点时，回溯到之前的节点进行访问。

②在广度优先搜索中，我们使用队列，从一个特定点开始，访问所有邻接的节点。然后对于这些邻接节点，我们依然通过访问邻接节点的方式，知道访问所有可以到达的节点。因此，我们按照一层一层的方式访问节点，

关键代码

①深度优先

    public static int findCircleNum(int[][] M) {
        int n=M.length;
        int []visited=new int[n];//保存是否遍历过,0为未遍历
        int count=0;//连通块的个数
        for(int i=0;i<n;i++){//初始化visited数组
            visited[i]=0;
        }

        for(int i=0;i<n;i++){
            if(visited[i]==0){
                DFS(M,visited,i);
                count++;
            }
        }
        return count;


    }

    public static void DFS(int[][] M,int []visited,int i){
        for (int j=0;j<visited.length;j++){
            if(M[i][j]==1&&visited[j]==0){//i与j是朋友且j未访问
                visited[j]=1;
                DFS(M,visited,j);//当前节点变为j并继续深度搜索

            }
        }
    }

②广度优先

    public static int findCircleNum(int[][] M){
        //值得注意的是LinkedList类实现了Queue接口，
        // 因此我们可以把LinkedList当成Queue来用。
        int n=M.length;
        int[] visited = new int[n];
        int count = 0;
        Queue< Integer > queue = new LinkedList< >();

        for(int i=0;i<n;i++){//初始化visited数组
            visited[i]=0;
        }

        for(int i=0;i<n;i++){
            if(visited[i]==0){
                queue.add(i);
                while (!queue.isEmpty()){
                    int s=queue.remove();
                    visited[s]=1;
                    for (int j=0;j<n;j++){
                        if(M[s][j]==1&&visited[j]==0)
                            queue.add(j);
                    }
                }
                count++;//当前队列无节点，证明连通块+1

            }
        }
        return count;


    }