算法学习之从上往下打印二叉树

题目描述

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

该题主要是考察队列Queue的使用类似于栈。函数poll()弹出头部值并去除，peek()函数弹出头部值但不会去除，offer()拼接值到队列尾部。LinkedList实现了Queue接口。

import java.util.*;
/**
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>();
        if(root==null){
           return list;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);
        while(queue.peek()!=null){
              TreeNode tem = queue.poll();
              list.add(tem.val);
              if(tem.left!=null)queue.offer(tem.left);
              if(tem.right!=null)queue.offer(tem.right);
        }
        return list;
    }
    
    
}

时间复杂度：O(n);

空间复杂度：O(n);