剑指offer系列（22）从上往下打印二叉树

题目描述

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

样例

思路分析

构造一个队列容器，即可轻松解决这道题

步骤	操作	队列
1	打印节点8	节点6、节点10
2	打印节点6	节点10、节点5、节点7
3	打印节点10	节点5、节点7、节点9、节点11
4	打印节点5	节点7、节点9、节点11
5	打印节点7	节点9、节点11
6	打印节点9	节点11
7	打印节点11

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

代码

方法一：直接利用队列

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

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.isEmpty()){
        	TreeNode temp = queue.poll();  //根节点出队
        	if (temp.left != null) {
				queue.offer(temp.left);
			}
        	if (temp.right != null) {
				queue.offer(temp.right);
			}
        	list.add(temp.val);
        }
        return list;    		
    }
	
	public class TreeNode {
	    int val = 0;
	    TreeNode left = null;
	    TreeNode right = null;
	    
	    public TreeNode(int val) {
	        this.val = val;
	    }
	}
}

结果

方法二：利用arraylist模拟队列操作

public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
		ArrayList<Integer> list = new ArrayList<Integer>();
        if (root == null){
        	return list;
        }
        ArrayList<TreeNode> queue = new ArrayList<TreeNode>();
        queue.add(root);
        while (!queue.isEmpty()){
        	TreeNode temp = queue.remove(0); //remove方法返回的是删除元素，在这里就是出队操作
        	if (temp.left != null) {
				queue.add(temp.left);
			}
        	if (temp.right != null) {
				queue.add(temp.right);
			}
        	list.add(temp.val);  //打印操作，每次一个
        }
        return list;  
    }

结果

利用arraylist的时间稍快于队列