树的广度优先遍历(BFS)-递归和迭代(Java)

树的广度优先遍历(BFS)-递归和迭代(Java)

## BFS

上面这棵树的广度优先遍历(层序遍历)的顺序是：

A–>B–>C–>D–>E–>F–>G–>H–>I

英文缩写BFS(Breadth First Search)。

其检验过程就是对每一层的节点的依次访问，访问完一层进入下一层，而且每个节点只能访问一次。

广度优先遍历树，需要用到队列（Queue）来存储节点对象，队列的特点就是先进先出。例如，上面这颗树的访问如下：

• 首先将A节点插入队列中，队列中有元素（A）;
• 将A节点弹出，同时将A节点的左、右节点依次插入队列，B在队首，C在队尾，（B，C），此时得到A节点；
• 继续弹出队首元素，即弹出B，并将B的左、右节点插入队列，C在队首，E在队尾（C,D，E），此时得到B节点；
• 继续弹出，即弹出C，并将C节点的左、中、右节点依次插入队列，（D,E,F,G,H），此时得到C节点；
• 将D弹出，此时D没有子节点，队列中元素为（E,F,G,H），得到D节点；
• 。。。以此类推。。

迭代递归

 public static List<Integer> bfs(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        List<Integer> list = new LinkedList<>();

        if (root == null) return list;
        queue.add(root);
        while (!queue.isEmpty()) {
            //将队列的头部的元素拿出来
            TreeNode tree = queue.remove();
            if (tree.left != null) queue.add(tree.left);
            if (tree.right != null) queue.add(tree.right);
            list.add(tree.val);
        }
        return list;
    }

 public List<TreeNode> traversal(TreeNode root) {
        if (root == null) return null;
        return bfs(root);
    }

    private List<TreeNode> bfs(TreeNode root) {
        int curNum = 1;
        int nextNum = 0;
        Queue<TreeNode> queue = new LinkedList<>();
        List<TreeNode> res = new ArrayList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            res.add(node);
            curNum--;
            if (node.left != null) {
                queue.add(node.left);
                nextNum++;
            }

            if (node.right != null) {
                queue.add(node.right);
                nextNum++;
            }

            if (curNum == 0) {
                curNum = nextNum;
                nextNum = 0;
            }
        }
        return res;
    }

    public List<TreeNode> Bfs_tree(TreeNode root){
        Queue<TreeNode> myq = new LinkedList<>();
        List<TreeNode> res = new ArrayList<>();
        if(root==null) return null;
        myq.add(root);
        while(!myq.isEmpty()){
            int len = myq.size();
            for(int i=0;i<len;i++){
                if(myq.peek().left!=null) myq.add(myq.peek().left);
                if(myq.peek().right!=null) myq.add(myq.peek().right);
                res.add(myq.poll());
            }
        }
        return res;
    }