JZ22 从上往下打印二叉树

原题链接

描述

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

示例1

输入：{5,4,#,3,#,2,#,1}
返回值：[5,4,3,2,1]

思路

层次遍历。

解答

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Queue;

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        if (root==null) return new ArrayList<>();


        Queue<TreeNode> queue = new ArrayDeque<>();
        ArrayList<Integer> arrayList = new ArrayList<>();

        queue.add(root);
        while (!queue.isEmpty()){
            TreeNode remove = queue.remove();
            arrayList.add(remove.val);
            if (remove.left != null) queue.add(remove.left);
            if (remove.right != null) queue.add(remove.right);
        }

        return arrayList;
    }
}