请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
提示:
节点总数 <= 1000
具体代码实现如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
// 双栈法,利用栈的先进后出特点,(没有下面那个方法快)
List<List<Integer>> res = new LinkedList<>();
Stack<TreeNode> odd = new Stack<>();// 奇数栈
Stack<TreeNode> even = new Stack<>();// 偶数栈
if(root == null) return res;// 非空判断
odd.push(root);// 将根节点压入奇数栈
int count = 1;// 用于计录层数
// 当奇数栈和偶数栈均为空时跳出
while(!odd.isEmpty() || !even.isEmpty()){
// 用于记录每层的结果
ArrayList<Integer> list = new ArrayList<>();
if(count % 2 == 1){// 奇数层
// 奇数层非空时
while(!odd.isEmpty()){
TreeNode cur = odd.pop();
list.add(cur.val);
// 将下一层的节点加入到偶数栈中,因为倒叙,刚好符合栈的先进后出特点
if(cur.left != null) even.push(cur.left);
if(cur.right != null) even.push(cur.right);
}
}else{// 偶数层
// 偶数层非空
while(!even.isEmpty()){
TreeNode cur = even.pop();
list.add(cur.val);
// 将下一层节点加入到奇数栈中,因为奇数层是正序,所以先加右节点
if(cur.right != null) odd.push(cur.right);
if(cur.left != null) odd.push(cur.left);
}
}
count++;
res.add(list);
}
return res;
/*
// 注意后面只需要new一个LinkeList<>();即可
List> res = new LinkedList<>();
// 创建队列,利用其先进先出的特点,对二叉树进行BFS遍历
Queue queue = new LinkedList<>();
// 判断根节点是否为空,注意,这里不能直接返回null
if(root == null) return res;
// 将根节点加入到队列中
queue.add(root);
// 用于反转时判断
boolean flag = true;
// 当队列为空时,跳出循环
while(!queue.isEmpty()){
// 因为要分层打印,故先回去每一层的长度
int len = queue.size();
// 因为要返回一个list集合的类型,故创建一个小的list,用于存储每一层的结果
List list = new ArrayList<>();
// 通过循环可遍历打印每一层的元素
for(int i = 0; i < len; i++){
TreeNode cur = queue.poll();
list.add(cur.val);
if(cur.left != null) queue.add(cur.left);
if(cur.right != null) queue.add(cur.right);
}
// 偶数行进行反转
if(!flag) Collections.reverse(list);
// 将结果接入到大的list集合中
res.add(list);
flag = !flag;
}
return res;
*/
}
}
人生若只如初见,何事秋风悲画扇。
等闲变却故人心,却道故人心易变。
-----------纳兰性德
小白寄语:学如逆水行舟,不进则退。