问题
Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).
Have you met this question in a real interview? Yes
Example
Given binary tree {3,9,20,#,#,15,7},
return its zigzag level order traversal as:
[
[3],
[20,9],
[15,7]
]
分析
使用递归能极大的简化操作。考虑是构造一个List来存储每一行的元素,然后用一个变量来记录本行是正序还是反序。
我们以上边的这个为例,第一行是正序,第二行就是反序,List存储的是2,3,我们在遍历第二行的时候需要把第一行加入结果集中,第一行我们正常遍历。但是我们希望第二行是7654,而我们是先遍历2,所以只能先取2的左结点,再取右结点,但是每次都是往队首插入。这样就是4, 54, 654,7654。
如果第一行是反序,第二行就是正序,List存储的是32,我们在遍历第二行的时候需要把第一行加入结果集中,第一行我们正常遍历。但是我们希望第二行是4567,而我们是先遍历3,所以只能先取3的右结点,再取左结点,但是每次都是往队首插入。这样就是7, 67, 567,4567。
如果我们修改遍历List的顺序也是可以的,方法类似。
代码
/**
* Definition of TreeNode:
* public class TreeNode {
* public int val;
* public TreeNode left, right;
* public TreeNode(int val) {
* this.val = val;
* this.left = this.right = null;
* }
* }
*/
public class Solution {
/*
* @param root: A Tree
* @return: A list of lists of integer include the zigzag level order traversal of its nodes' values.
*/
public List> zigzagLevelOrder(TreeNode root) {
// write your code here
List> res=new ArrayList();
if(root!=null){
List temp=new ArrayList();
temp.add(root);
treeWalk(res,temp,true);
}
return res;
}
private void treeWalk(List> res,List temp,boolean isLeft2Right){
List list=new ArrayList();
List temp2=new ArrayList();
for(TreeNode node:temp){
list.add(node.val);
if(isLeft2Right){
if(node.left!=null){
temp2.add(0,node.left);
}
if(node.right!=null){
temp2.add(0,node.right);
}
}else{
if(node.right!=null){
temp2.add(0,node.right);
}
if(node.left!=null){
temp2.add(0,node.left);
}
}
}
res.add(list);
temp=null;
if(temp2.size()>0){
treeWalk(res,temp2,!isLeft2Right);
}
}
}