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

概述

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

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[20,9],[15,7]]

示例 2：

输入：root = [1]
输出：[[1]]

示例 3：

输入：root = []
输出：[]

提示：

· 树中节点数目在范围 [0, 2000] 内
· -100 <= Node.val <= 100

解题思路

使用双端队列LinkedList，在从左往右层从队列头部依次弹出节点，并以左子节点先，右子节点后的顺序插入到队列尾部；在从右往左层从队列尾部依次弹出节点，并以右子节点先，左子节点后的顺序插入到队列头部。

解题代码

/**
 * 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) {
        List<List<Integer>> res = new ArrayList<>();
        if(root==null)return res;
        Deque<TreeNode> queue=new LinkedList<>();
        queue.add(root);
        int count=0;
        while (!queue.isEmpty()){
            int n=queue.size();
            List<Integer> tmpList=new ArrayList<>();
            if (count%2==0){
                for (int i=0;i<n;i++){
                    TreeNode tmpNode1=queue.pollFirst();
                    tmpList.add(tmpNode1.val);
                    if (tmpNode1.left!=null)queue.addLast(tmpNode1.left);
                    if (tmpNode1.right!=null)queue.addLast(tmpNode1.right);
                }
            }else {
                for (int j=0;j<n;j++){
                    TreeNode tmpNode2=queue.pollLast();
                    tmpList.add(tmpNode2.val);
                    if (tmpNode2.right!=null)queue.addFirst(tmpNode2.right);
                    if (tmpNode2.left!=null)queue.addFirst(tmpNode2.left);
                }
            }
            count++;
            res.add(tmpList);
        }
        return res;
    }
}