剑指offer——从上往下打印二叉树

import java.util.ArrayList;
/**
* @author Tom Qian
* @email tomqianmaple@outlook.com
* @github https://github.com/bluemapleman
* @date 2017年8月18日
*/
//要求：从上往下打印出二叉树的每个节点，同层节点从左至右打印。

//思路：典型的二叉树递归问题。

/*
这里我们可以借助队列结构来完成。首先我们用让一棵树的根节点入队，然后看它的左右子节点有没有，如果有左子节点，就入队，没有就直接看右节点。同理，右节点存在时就入队。接着就可以开始循环，每次把新出队的一个根节点加入到输出列表，并对其进行上述的处理。
*/

public class 从上往下打印二叉树
{
    public static ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
    ArrayList<Integer> list=new ArrayList<Integer>();
    if(root==null)
        return list;

    Queue<TreeNode> queue=new LinkedList<TreeNode>();


    queue.add(root);

    while(queue.size()!=0){
        root=queue.poll();
        list.add(root.val);
        if(root.left!=null){
            queue.add(root.left);
        }
        if(root.right!=null){
            queue.add(root.right);
        }
    }
    return list;
}

    public static void main(String[] args)
    {
        TreeNode node1,node2,node3,node4,node5,node6;
        node1=new TreeNode(1);
        node2=new TreeNode(2);
        node3=new TreeNode(3);
        node4=new TreeNode(4);
        node5=new TreeNode(5);

        node1.left=node2;
        node1.right=node5;
        node2.left=node3;
        node3.right=node4;
        ArrayList<Integer> list=PrintFromTopToBottom(node1);
        for(Integer ele:list)
            System.out.println(ele);
    }

}