Java实现LeetCode103.二叉树的锯齿层序遍历

题目描述

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

思路

这题偷懒了，直接查List怎么reverse，但又感觉复杂度高。看了题解发现可以直接利用双端队列的addFirst的方法，需要额外借助一个order判断当前是奇数行还是偶数行。

此外，看网友的评论，这种思路本质上还是层序遍历，不过在返回值的设计add上进行了一些正序或倒序的判断。

代码

/**
 * 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>> ans = new LinkedList();
        if( root==null ) return ans;
        boolean order = true;
        Deque<TreeNode> dq = new LinkedList();
        dq.add( root );
        while( !dq.isEmpty() ){
            int len = dq.size();
            List<Integer> tmp = new LinkedList();
            while( len > 0 ){
                TreeNode tmpNode = dq.poll();
                if( !order ) tmp.addFirst( tmpNode.val );
                else tmp.add( tmpNode.val );
                if( tmpNode.left!=null ) dq.add( tmpNode.left );
                if( tmpNode.right!=null ) dq.add( tmpNode.right );
                len--;
            }
            order = !order;
            ans.add( new LinkedList( tmp ) );
        }
        return ans;
    }
}