32.从上到下打印二叉树

32. 从上到下打印二叉树

1 题目描述

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

2 题目分析

​ 用队列就能简单实现，层序遍历

3 代码

public int[] levelOrder(TreeNode root) {
    if (root == null) return new int[0];
    List<Integer> ans = new ArrayList<>();

    // 队列用于层序遍历
    Deque<TreeNode> queue = new LinkedList<>();
    // 首节点入队
    queue.offer(root);
    while (!queue.isEmpty()) {
        TreeNode node = queue.poll();
        ans.add(node.val);
        // 入队
        if (node.left != null) queue.offer(node.left);
        if (node.right != null) queue.offer(node.right);
    }
    int size = ans.size();
    int[] ret = new int[size];
    for (int i = 0; i < size; i++) {
        ret[i] = ans.get(i);
    }
    return ret;
}

4 两种变形

1. 分行从上到下打印二叉树

// 2. 按层打印：打印前先返回当前层的节点数
public List<List<Integer>> levelOrder1(TreeNode root) {
    List<List<Integer>> ans = new ArrayList<>();
    if (root == null) return ans;

    Deque<TreeNode> queue = new LinkedList<>();
    List<Integer> curLevel;
    // 根节点入队
    queue.offer(root);
    while (!queue.isEmpty()) {
        int size = queue.size();  // 当前层的大小
        curLevel = new ArrayList<>();
        for (int i = 0; i < size; i++) {
            // 出队
            TreeNode node = queue.pop();
            curLevel.add(node.val);
            if (node.left != null)  queue.offer(node.left);
            if (node.right != null) queue.offer(node.right);
        }
        ans.add(curLevel);
    }
    return ans;
}

2. 之字形打印二叉树：用一个双端队列去保存每层的节点（Deque的用法可见另一篇文章：Deque相关方法详解）

// 3. 之字形打印:用一个辅助栈来控制
public List<List<Integer>> levelOrder2(TreeNode root) {
    List<List<Integer>> ans = new ArrayList<>();
    if (root == null) return ans;

    Deque<TreeNode> queue = new LinkedList<>();
    List<Integer> curLevel;
    boolean isLeftToRight = true;
    // 根节点入队
    queue.offer(root);
    while (!queue.isEmpty()) {
        Deque<Integer> level = new LinkedList<>();
        int size = queue.size();  // 当前层的大小
        curLevel = new ArrayList<>();
        for (int i = 0; i < size; i++) {
            // 出队
            TreeNode node = queue.pop();
            // curLevel.add(node.val);
            if (isLeftToRight) {
                level.offerLast(node.val);
            } else {
                level.offerFirst(node.val);
            }
            if (node.left != null)  queue.offer(node.left);
            if (node.right != null) queue.offer(node.right);
        }
        isLeftToRight = !isLeftToRight;
        ans.add(new ArrayList<Integer>(level));
    }
    return ans;
}