朋友圈(力扣第547题)

题目:

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

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

示例:

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

注意:

N 在[1,200]的范围内。
对于所有学生,有M[i][i] = 1。
如果有M[i][j] = 1,则有M[j][i] = 1

分析:

  这个给定的二维矩阵其实就是图的邻接矩阵的表示形式,根据题目分析朋友关系是具有传递性的,所以可以判断是无向图,N*N代表着这个无向图中共有N个顶点,这些个顶点中是否存在边,就看邻接矩阵中对应位置的元素值是否为1,如果为1则说明是有边的,如果值为0,则说明顶点之间没有边。至于求朋友圈的个数,其实就是求这个无向图中连通分量的个数,也就是一共有多少个子图,要注意的是即使只有一个顶点,那也可以构成一个子图。那么我们就通过深度优先遍历算法,然后设置一个访问标记数组,寻找这个图中一共有多少个连通分量。

代码:

 

public class Tip547 {
    private int n;
    public int findCircleNum(int[][] M) {
        if (M == null){
            return 0;
        }

        n = M.length;
        int fri_total = 0;

        boolean[] isVisited = new boolean[n];

        for (int i = 0; i < n; i++) {

            if (!isVisited[i]){

                findFriByDFS(M, i, isVisited);
                fri_total++;
            }
        }

        return fri_total;

    }

    private void findFriByDFS(int[][] M,int i,boolean[] isVisited){

        isVisited[i] = true;
        for (int j = 0; j < n; j++) {

            if (M[i][j]==1 && !isVisited[j]){
                findFriByDFS(M,j,isVisited);
            }
        }

    }

    public static void main(String[] args) {

        Tip547 tip547 = new Tip547();
        int[][] M = {{1,0,0,1},{0,1,1,0},{0,1,1,1},{1,0,1,1}};

        System.out.println(tip547.findCircleNum(M));
    }
}

 

  其实DFS的算法实现了套路无非就是,确定好哪些是图中的顶点,然后从起始顶点开始深度遍历时,要找好从每个点向下深度遍历时它拥有的”走法“,这些”走法“往往构成一个for循环,目的是为了走某个方向时,走到尽头,然后再退回来,走其他方向。在本题中,矩阵就是所有顶点下一步”走法“的合集,每一维度都是某一个顶点的所有可走的选择,当从某一个顶点i进入深度遍历函数时,会进行for循环操作,一旦有M[i][j]等于1并且尚未被访问,那就开始以j为起点进行下一步的深度遍历。走到尽头,就向上返回,继续循环,直到走完从最先进入DFS函数的那个点出发的可以走通的极大连通子图。

  然后因为DFS遍历时图中的每个顶点只能访问依次,所以要设置一个访问标记数组,这个数组用于记录顶点是否被访问,防止重复访问。

 

posted @ 2020-06-19 20:29  有心有梦  阅读(430)  评论(0编辑  收藏  举报