题目描述
请实现一个函数按照之字形打印二叉树,根节点之后的第一层按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
/**
* 8
* / \
* 6 10
* / \ / \
* 5 7 9 11
* @param pRoot
* @return
*/
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer> > result = new ArrayList<ArrayList<Integer> >();
if(pRoot == null){
return result;
}
boolean flg = true;
//根节点 为 之字的点 不算第一层
int k = 0;
ArrayList<Integer> list = null;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(pRoot);
while(!queue.isEmpty()){
int len = queue.size();
list = new ArrayList<Integer>();
for(int i=0;i<len;i++){
TreeNode temp = queue.poll();
if(temp.left!=null){
queue.offer(temp.left);
}
if(temp.right!=null){
queue.offer(temp.right);
}
if(flg){
list.add(0,temp.val);
}else{
list.add(temp.val);
}
}
if(k > 0){
flg = flg ? false : true;
}
k++;
result.add(list);
}
return result;
}
}
打印之 图形结果:
[8]
[10,6]
[5,7,9,11]