从上往下打印二叉树 java

版权声明：博客内容为本人自己所写，请勿转载。 https://blog.csdn.net/weixin_42805929/article/details/83107180

从上往下打印二叉树 java

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

思路：
按照层次遍历的方法，使用队列辅助。
1.将根结点加入队列。
2.循环出队，打印当前元素，若该结点有左子树，则将其加入队列，若有右子树，将其加入队列。
3.直到队列为空，表明已经打印完所有结点。

代码1：

import java.util.*;

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> al = new ArrayList<>();
        ArrayList<TreeNode> queue = new ArrayList<>();
        if(root == null){
            return al;
        }
        queue.add(root);
        while(queue.size() != 0){
            TreeNode temp = queue.remove(0);
            al.add(temp.val);
            if(temp.left != null){
                queue.add(temp.left);
            }
            if(temp.right != null){
                queue.add(temp.right);
            }
        }
        return al;
    }
}

代码2：

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> al = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        if(root == null){
            return al;
        }
        queue.offer(root);
        while(!queue.isEmpty()){
            TreeNode temp = queue.poll();
            if(temp.left != null){
                queue.offer(temp.left);
            }
            if(temp.right != null){
                queue.offer(temp.right);
            }
            al.add(temp.val);
        }
        return al;
    }
}