剑指offer_从上往下打印二叉树

题目描述

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

例如，以下二叉树层次遍历的结果为：1,2,3,4,5,6,7

 

 

解题思路就是使用一个队列，第一次入队根节点，之后一直循环直到队列为空，每次弹出队列第一个元素并添加第一个元素的左右节点。

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    ArrayList<TreeNode> arr = new ArrayList<>();
    ArrayList<Integer> ans = new ArrayList<>();
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        if(root == null) return ans;
        arr.add(root);
        while(!arr.isEmpty()){
            TreeNode memo = arr.remove(0);
            ans.add(memo.val);
            if(memo.left!=null) arr.add(memo.left);
            if(memo.right!=null) arr.add(memo.right);
            
        }
        return ans;
    }
}