力扣第103题-二叉树的锯齿形层序遍历

题目
链接

题解

import java.util.List;
import java.util.ArrayList;
import java.util.Deque;
import java.util.ArrayDeque;

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>(0);
        }
        List<List<Integer>> res = new ArrayList<>();
        // 将双端队列分成两个栈使用，一端弹栈时，另一端压栈
        Deque<TreeNode> deque = new ArrayDeque<>();
        deque.addLast(root);
        // 定义一个时序标记位
        // true:  addFirst removeLast  添加时 从右往左 即 先添加右子树 再添加左子树
        // false: addLast  removeFirst 添加时 从左往右 即 先添加左子树 再添加右子树
        boolean seq = false;
        while (!deque.isEmpty()) {
            int size = deque.size();
            List<Integer> floor = new ArrayList<>(size);
            for (int i = 0; i < size; i++) {
                TreeNode node = seq ? deque.removeLast() : deque.removeFirst();
                floor.add(node.val);

                // 从左往右 从此处开始执行
                if (node.left != null && !seq) {
                    add(deque, node.left, seq);
                }

                // 从右往左 从此处开始执行
                if (node.right != null) {
                    add(deque, node.right, seq);
                }

                // 从左往右 需要在此处终止
                if (!seq) {
                    continue;
                }

                if (node.left != null) {
                    add(deque, node.left, seq);
                }
            }
            // 一层结束
            res.add(floor);
            // 反转时序
            seq = !seq;
        }
        return res;
    }

    // 辅助方法，方便入队
    private void add(Deque<TreeNode> deque, TreeNode e, boolean seq) {
        if (seq) {
            deque.addFirst(e);
        } else {
            deque.addLast(e);
        }
    }

}