请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路
在之前的基础上,借用了栈和两个队列。
解法
nowcoder的AC
提交时间:2020-01-10 ,语言:Java ,运行时间: 30 ms ,占用内存:9616K ,状态:答案正确
import java.util.*;
/*
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) {
ArrayList<ArrayList<Integer> > list1 = new ArrayList<>();
ArrayList<Integer> list2 = new ArrayList<>();
if(pRoot == null){
return list1;
}
list1.add(list2);
Queue<TreeNode> queue1 = new LinkedList<>();
Queue<TreeNode> queue2 = new LinkedList<>();
queue1.offer(pRoot);
TreeNode node;
Stack<TreeNode> stack = new Stack<>();
int c=1;
Queue<TreeNode> q = null;
while(!queue1.isEmpty() || !queue2.isEmpty() ){
if(queue1.isEmpty()){
c++;
stack = new Stack<>();
queue1 = new LinkedList<>(queue2);
q=new LinkedList<>(queue1);
queue2.clear();
list2 = new ArrayList<>();
list1.add(list2);
}
if(c%2==0){
while (!q.isEmpty()){
stack.push(q.poll());
}
list2.add(stack.pop().val);
node=queue1.poll();
}else{
node = queue1.poll();
list2.add(node.val);
}
if(node.left != null){
queue2.offer(node.left);
}
if(node.right != null){
queue2.offer(node.right);
}
}
return list1;
}
}