剑指Offer-Java-从上往下打印二叉树

package com.sjsq.test;

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;

/**
 * @author shuijianshiqing
 * @date 2020/5/24 10:43
 */

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

public class Solution {

    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {

        // 定义双向队列
        Deque<TreeNode> deque = new LinkedList<>();
        // 定义数组来存放结果
        ArrayList<Integer> result = new ArrayList<>();

        if(root == null){
            return result;
        }

        deque.add(root);

        while(!deque.isEmpty()){
            TreeNode node = deque.getFirst();
            deque.pollFirst();
            result.add(node.val);

            // 判断有没有左子树
            if(node.left != null){
                deque.addLast(node.left);
            }

            // 判断有没有右子树
            if(node.right != null){
                deque.addLast(node.right);
            }
        }

        return result;
    }
}