Java从上往下打印二叉树

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

规律:每一次打印一个节点的时候,如果该节点有子节点,则把该节点的子节点放到一个队列的尾部。接下来到对队列的头部取出最早进入队列的节点放到ArrayList 中,重复前面的操作,直至队列中所有的节点都存到ArrayList中。


/**
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) {
        //queue用来保存当前遍历到了哪个节点,一次性把一个节点的左右子都入队
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        //list用来保存输出的节点
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(root==null){//注意:空树返回一个默认构造的空LinkedList,而不是一个空指针null
            return list;
        }
        TreeNode current = root;
        queue.offer(current);
        //只要队列中还有节点就说明还没遍历完,继续。
        //每次从队列出队,然后将这个节点左右子入队列(FIFO,故能完成广度/层级遍历),再将这个节点记录在list中即可。
        while(!queue.isEmpty()){
            current = queue.poll();
            list.add(current.val);
            if(current.left!=null){//有左子则入队
                queue.offer(current.left);
            }
            if(current.right!=null){
                queue.offer(current.right);
            }
        }
        return list;
    }
}
发布了45 篇原创文章 · 获赞 50 · 访问量 2037

猜你喜欢

转载自blog.csdn.net/zhaoyi666/article/details/105423564
今日推荐