题目:给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)。
思路:此题是层序遍历基础的不断演化。首先是写入层序遍历的代码:利用队列。然后通过count变量控制每次循环的次数,从而将每层的数据都分别保存到一个list上,最后引入一个num标志位,来控制奇偶层,如果为奇数层list顺序不变,如果是偶数层就将链表list中的数据反转。
import java.util.*;
import java.util.ArrayList;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
if(pRoot == null) return new ArrayList();
Queue<TreeNode> queue = new LinkedList();
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
queue.add(pRoot);
int num = 1;
while(!queue.isEmpty()){
ArrayList<Integer> list = new ArrayList();
int count = queue.size();
for(int i=0;i<count;i++){
TreeNode node = queue.remove();
if(node.left != null){
queue.add(node.left);
}
if(node.right != null){
queue.add(node.right);
}
list.add(node.val);
}
if(num%2==0){
for(int m=0,n=list.size()-1;m<n;m++,n--){
int tmp = list.get(m);
list.set(m,list.get(n));
list.set(n,tmp);
}
}
num++;
res.add(list);
}
return res;
}
}