Leetcode 261. 以图判树

问题描述

java实现

class Solution {
        public boolean validTree(int n, int[][] edges) {
            if(n==0)
                return false;
            // 判断是否满足条件：1. 刚好N-1条边
            if(edges.length!=n-1)
                return false;
            Map<Integer, Set<Integer>> graph=initializeGraph(n,edges);

            //bfs
            Queue<Integer> queue=new LinkedList<>();
            Set<Integer> hash=new HashSet<>();//所有进去过的点

            queue.offer(0); //从0出发 把0加入队列
            hash.add(0);
            while (!queue.isEmpty()){
                int node=queue.poll();//去除队列第一个
                for(Integer neighbor:graph.get(node)){ //找所有的邻居
                    if(hash.contains(neighbor))//重复判断
                        continue;//有重复
                    hash.add(neighbor);
                    queue.offer(neighbor);
                }
            }
            return (hash.size()==n);
        }

        //邻接表  map 键值对中 键为某一个点 值为这个点所有的邻居(用set存储)
        private Map<Integer,Set<Integer>> initializeGraph(int n,int[][] edges){
            Map<Integer,Set<Integer>> graph=new HashMap<>();
            for (int i = 0; i <n ; i++)
                graph.put(i,new HashSet<Integer>());
            for (int i = 0; i <edges.length ; i++) {
                int u=edges[i][0];
                int v=edges[i][1];
                graph.get(u).add(v);
                graph.get(v).add(u);
            }
            return graph;
        }
    }