《剑指 Offer》——59、按之字形顺序打印二叉树

1. 本题知识点

树

2. 题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

3. 解题思路

1. 创建一个队列。创建一个线性表，其中一个元素是一层二叉树。
2. 将二叉树的根节点加入队尾
3. 设置一个标志为 false
4. 当队列不为空时，执行以下循环
   1. 记录本层二叉树的结点数
   2. 当本层二叉树的结点数大于 0，执行以下循环
      1. 删除队头
      2. 将删除的队头加入本层二叉树的线性表
      3. 将删除的队头的左右子树加入队尾
      4. 本层二叉树结点树减 1
   3. 当标志为 true 时，则将线性表顺序反转
   4. 标志反转
   5. 将本层二叉树的线性表加入线性表
5. 返回线性表

4. 代码

public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }

}

public class Solution {
    /**
     * 从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。
     *
     * @param pRoot
     * @return
     */
    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        // 创建一个队列
        Queue<TreeNode> queue = new LinkedList<>();
        // 创建一个线性表，其中一个元素是一层二叉树
        ArrayList<ArrayList<Integer>> lists = new ArrayList<>();
        // 根节点不为空时，将其加入队尾
        if (pRoot != null) {
            queue.add(pRoot);
        }
        // 是否将线性表顺序反转
        Boolean flag = false;
        // 队列不为空时，执行循环
        while (!queue.isEmpty()) {
            ArrayList<Integer> list = new ArrayList<>();
            // 本层二叉树的结点数
            int count = queue.size();
            // 本层二叉树的结点数大于 0，执行循环
            while (count > 0) {
                // 删除队头
                TreeNode removeNode = queue.remove();
                // 将删除的队头加入线性表
                list.add(removeNode.val);
                // 如果删除的队头有左子树，将其加入队尾
                if (removeNode.left != null) {
                    queue.add(removeNode.left);
                }
                // 如果删除的队头有右子树，将其加入队尾
                if (removeNode.right != null) {
                    queue.add(removeNode.right);
                }
                // 本层二叉树结点树减 1
                count--;
            }
            if (flag) {
                // 线性表顺序反转
                Collections.reverse(list);
            }
            flag = !flag;
            // 将本层二叉树加入线性表
            if (list.size() != 0) {
                lists.add(list);
            }
        }
        return lists;
    }
}