剑指offer32-1：从上到下打印二叉树

1. 题目描述

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

输出：[1, 2, 3, 4, 5]

2. 解题思路

其实为二叉树的层序遍历（广度优先）。

①第一个节点为1，存储到容器1中，list = [1]

②将容器中第一个元素取出，放进最终答案的容器2中，ans = [1]；

并将刚才取出元素的左右节点依次放入一开始的容器1，list = [2, 3]

③重复第二个步骤：

        list = [3]            ans = [1, 2]

        list = [4, 5]        ans = [1, 2, 3]

        list = [5]            ans = [1, 2, 3, 4]

        list = []              ans = [1, 2, 3, 4, 5]

我们发现，容器1有着先进先出的特点，即为一个队列。

3. 代码

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

public class ques32 {
    public static int[] levelOrder(TreeNode root) {
        if(root == null) return new int[0];
        int[] res;
        //列表
        List<Integer> list = new ArrayList<>();
        //队列：用来存储树节点的值
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        //直到队列为空为止
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();  //返回并移除队列的头元素
            list.add(node.val);
            if(node.left != null)
                queue.add(node.left);
            if(node.right != null)
                queue.add(node.right);
        }

        //输出结果
        int[] arr = new int[list.size()];
        for(int i = 0; i < arr.length; i ++)
            arr[i] = list.get(i);

        return arr;

        //人家的方式
        //Integer[] arr = list.toArray(new Integer[list.size()]);
        //res = new int[arr.length];
        //for(int i = 0; i < arr.length; i ++)
        //    res[i] = arr[i];
        //return res;
    }

    public static class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode(int x) { val = x; }
    }

    public static void main(String[] args) {
        //构建树
        TreeNode a4 = new TreeNode(4);
        TreeNode a5 = new TreeNode(5);
        TreeNode a3 = new TreeNode(3);
        a3.left = a4;
        a3.right = a5;
        TreeNode a2 = new TreeNode(2);
        TreeNode a1 = new TreeNode(1);
        a1.left = a2;
        a1.right = a3;

        //测试
        int[] res = levelOrder(a1);
        for (int i = 0; i < res.length; i++) {
            System.out.println(res[i]);
        }
    }
}