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

最新推荐文章于 2020-07-19 21:10:28 发布

原创最新推荐文章于 2020-07-19 21:10:28 发布 · 204 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

65 篇文章

订阅专栏

博客介绍了从上往下打印二叉树节点（即层序遍历）的方法。借助队列，先将根节点入队，再依次取出节点打印，若节点有子节点则将子节点入队，循环操作直至队列为空，还给出了参考资料《java程序员面试宝典》（第四版）。

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

思路：实际就是层序遍历二叉树。可以借助队列，先将根节点放入队列中，然后依次从队列中取出节点打印。若该结点有子结点，则把子结点放进队列，依次循环，直到队列为空。

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/**
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<Integer>();        
        Queue<TreeNode> queue=new LinkedList<TreeNode>();
        if(root==null){
            return list;//因为List被初始化为[],不可以直接返回null
        }
        queue.add(root);
        while(!queue.isEmpty()){
            if(queue.peek().left!=null){
                queue.add(queue.peek().left);
            }
            if(queue.peek().right!=null){
                queue.add(queue.peek().right);
            }
            list.add(queue.poll().val);
        }
        return list;
    }
}

参考资料：《java程序员面试宝典》（第四版）