【編程題】從上往下打印二叉樹（Java實現）

【編程題】從上往下打印二叉樹（Java實現）

題目來源

劍指offer第22題
https://www.nowcoder.com/questionTerminal/7fe2212963db4790b57431d9ed259701

題目描述

從上往下打印出二叉樹的每個節點，同層節點從左至右打印。

題目代碼

import java.util.*;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer>list=new ArrayList<>();
        if(root==null)
            return list;
        LinkedList<TreeNode>lay=new LinkedList<>();
        lay.add(root);
        while(!lay.isEmpty()){
            root=lay.poll();
            if(root.left!=null)
               lay.add(root.left);
            if(root.right!=null)
               lay.add(root.right);
            list.add(root.val);
        }
        return list;
    }
}