题目描述

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

题解

利用队列进行简单的层次遍历即可。

public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
    ArrayList<Integer> res = new ArrayList<>();
    Queue<TreeNode> queue = new LinkedList<>();

    if (root == null)
        return res;
    queue.offer(root);

    while (!queue.isEmpty()) {
        TreeNode p = queue.poll();
        res.add(p.val);
        if (p.left != null)
            queue.offer(p.left);
        if (p.right != null)
            queue.offer(p.right);
    }
    return res;
}