从上往下打印二叉树

题目描述

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

思路

1. 借助一个ArrayList充当队列的角色。
   2.每次从队列头部取出元素后，分别判断其左右孩子，若不为空就加入到队尾，若为空则跳过。
   3.当队列中没有元素时，整个二叉树就遍历完成了。

Java代码实现

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        if(root == null){
            return new ArrayList<Integer>();
        }
        ArrayList<Integer> array = new ArrayList<Integer>();
        ArrayList<TreeNode> queue = new ArrayList<TreeNode>();
        
        queue.add(root);
        
        while(queue.size()>0){
            TreeNode cur = queue.remove(0);
            
            array.add(cur.val);
            if(cur.left != null){
                queue.add(cur.left);
            }
            
            if(cur.right != null){
                queue.add(cur.right);
            }
        }
        
        return array;
    }
}