【LeetCode(Java) - 261】以图判树

1、题目描述

在这里插入图片描述

2、解题思路

  只要存在环,就不是一个数。

进行 BFS 遍历。

定义一个二维数组 graph[][] 用来存储边的信息,如 graph[2][7] 表示结点 2 和结点 7 之间存在边。

定义一个一维数组 visited[] 用来存储结点已处理信息,如 visited[3] = true,表示结点 3 已经处理过。

从结点 0 开始进行 BFS,假设当前遍历到的结点为 cur:

1、visited[cur] = true 标记当前结点已处理;

2、找出 cur 的所有邻接结点;

3、依次把 cur 和邻接结点的边都给去掉,同时把邻接结点也设置为 true。

如果在第 3 步之前,发现 cur 的一个邻接结点的 visited[naborNode] 已经为 true ,说明存在环。

解释:如果 visited[naborNode] 为 true,说明 naborNode 是前面某个结点的邻接节点,而 cur 也是前面某个节点的邻接节点,加上 cur 和 naborNode 又是连在一起的,于是形成了环。
在这里插入图片描述
  BFS 结束后,如果是合法的树结构,此时应该已经处理完所有节点,如果还存在某个节点 visited[node] == false,说明连通分量不为 1。

3、解题代码

class Solution {
    
    
    public boolean validTree(int n, int[][] edges) {
    
    
        //构建邻接矩阵
        int[][] graph = new int[n][n];
        //有边的元素设置为1,没有边的元素设置为0
        for (int[] edge : edges) {
    
    
            // graph[3][4] == 1 表示 3 和 4 是连接的
            graph[edge[0]][edge[1]] = 1;
            graph[edge[1]][edge[0]] = 1;
        }
        //进行BFS
        Queue<Integer> queue = new LinkedList<>();
        //从第一个节点开始搜索,这样就不会漏掉无边图的情况
        queue.add(0);
        boolean[] visited = new boolean[n];
        while (!queue.isEmpty()) {
    
    
            Integer cur = queue.poll();
            visited[cur] = true;
            //获取邻接点
            for (int i = 0; i < n; i++) {
    
    
                //查看当前节点的邻接点
                if (graph[cur][i] == 1) {
    
    
                    if (visited[i]) {
    
    
                        // cur 的邻接节点居然被处理过
                        // 说明 cur 和 i 在前面有一个共同的父结点
                        // 加上 cur 和 i 又是连在一起的
                        // 说明存在环
                        return false;
                    }
                    visited[i] = true;
                    //涂黑访问过的节点
                    graph[cur][i] = 0;
                    graph[i][cur] = 0;
                    queue.add(i);
                }
            }
        }

        //判断是否为单连通分量
        for (int i = 0; i < n; i++) {
    
    
            if (!visited[i]) {
    
    
                // 居然还有一个节点没有被访问过,说明不是图
                return false;
            }
        }
        return true;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_29051413/article/details/108531255