题目描述
给你二叉树的根节点 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;
}
}