[算法练习-剑指offer]题22.从上往下打印二叉树(层序遍历二叉树)（Java）

题目

题号:22
题目名:从上往下打印二叉树(层序遍历二叉树)

编程语言

Java

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

初次思路

非常经典的广度优先遍历BFS

层序遍历,使用队列

解题代码

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        /**
         * 看题目描述.很明显就是层级打印
         * 层级打印使用bfs
         * bfs使用队列
         */
        ArrayList<Integer> res = new ArrayList<> ();
        //非空校验
        if(root ==null) return res;
        //经典BFS广度优先遍历
        Deque<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            int size = queue.size();
            //将队列中原有的结点加入res,同时加入左右节点
            while(size>0) {
                TreeNode node = queue.removeFirst();//出队
                res.add(node.val);
                if(node.left!=null) queue.addLast(node.left);//如果出队的结点有左节点则将左节点入队
                if(node.right!=null) queue.addLast(node.right);//如果出队的结点有右节点则将右节点入队
                size--;
            }
        }
        return res;
    }