题目链接:JZ77 按之字形顺序打印二叉树
题目描述:给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
数据范围:0
≤
\le
≤ n
≤
\le
≤ 1500,树上每个节点的val满足∣val∣<=1500
要求:空间复杂度:O(n),时间复杂度:O(n)
题目分析:层次遍历+记录层数,利用双端队列层次遍历的基础上,记录层数layer,如果为奇数层,则从左到右打印,偶数层,则从右向左打印。时间复杂度为O(n),空间复杂度也为O(n)
import java.util.*;
public class Solution {
public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> ret = new ArrayList<ArrayList<Integer>>();
if(pRoot==null) return ret;
//创建双端队列,用于保存出队结点的左右子结点
//Deque接口有两个实现类:ArrayDeque和LinkedList
Deque<TreeNode> deque = new LinkedList<TreeNode>();
deque.addLast(pRoot);
int layer = 1; //记录当前层数
while(!deque.isEmpty()){
int size = deque.size(); //记录当前层的结点个数
ArrayList<Integer> listTemp = new ArrayList<>();
if(layer%2==1){
//奇数层,从左到右打印
while(size!=0){
TreeNode topNode = deque.removeFirst(); //从队头出
listTemp.add(topNode.val); //将值添加到数组中
if(topNode.left!=null) deque.addLast(topNode.left); //先添加出队元素的左结点
if(topNode.right!=null) deque.addLast(topNode.right); //添加出队元素的右结点
size--;
}
layer++;
}else{
//偶数层,从右向左打印
while(size!=0){
TreeNode topNode = deque.removeLast(); //从队尾出
listTemp.add(topNode.val); //将值添加到数组中
if(topNode.right!=null) deque.addFirst(topNode.right); //先添加出队元素的右结点
if(topNode.left!=null) deque.addFirst(topNode.left); //添加出队元素的左结点
size--;
}
layer++;
}
ret.add(listTemp); //加入返回列表中
}
return ret;
}
}
总结:打印/遍历二叉树,要往前中后序遍历以及层序遍历上靠拢,然后更改限制条件;同时也要注意返回值的类型。选择合适的数据结构进行存储。多画图,多顺思路。加油鸭!