103. 二叉树的锯齿形层序遍历

103. 二叉树的锯齿形层序遍历

给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回锯齿形层序遍历如下：

[
  [3],
  [20,9],
  [15,7]
]

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        if(root == null) return new ArrayList<>();

        Stack<TreeNode> fistLeft = new Stack<>();
        Stack<TreeNode> fistRight = new Stack<>();
        List<List<Integer>> res = new ArrayList<>();

        fistLeft.add(root);
        while(!fistLeft.empty()||!fistRight.empty()){
            List<Integer> list = new ArrayList<>();
            while(!fistLeft.isEmpty()){
                TreeNode node = fistLeft.pop();
                if(node.left!=null) fistRight.push(node.left);
                if(node.right!=null) fistRight.push(node.right);
                list.add(node.val);
            }
            if(!list.isEmpty()){
                res.add(list);
                list = new ArrayList<>();
            }
            while(!fistRight.isEmpty()){
                TreeNode node = fistRight.pop();
                if(node.right!=null) fistLeft.push(node.right);
                if(node.left!=null) fistLeft.push(node.left);
                list.add(node.val);
            }
            if(!list.isEmpty()){
                res.add(list);
            }
        }

        return res;
    }
}

也可以用一个queue存储，size计数每行的长度，每次从队列中取出 \textit{size}size 个元素进行拓展，然后进行下一次迭代。对当前层节点的存储我们维护一个变量 \textit{isOrderLeft}isOrderLeft 记录是从左至右还是从右至左的