每日简短一刷,保持手感系列
题目:请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:给定二叉树: [3,9,20,null,null,15,7]
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
这题是这个系列的最后一题了,类似 Z 字形状打印,按照题意其实思路是很清晰的,关键就在于如何把思路转化为代码了。
毋庸置疑这个题目还是可以用队列来实现的,但是由于这个出队的顺序每一层不一致,那脑子里应该能蹦出双端队列,这样头尾都能出结点。
然后可以定义一个变量,用它的正负来判断这一层到底是哪一端出队,这样一来不就完美了?每遍历一层更改一下这个变量的正负即可。
当然,每一层入队的顺序也是不一样的,详情来看下代码应该就清晰了。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList();
if (root == null) {
return res;
}
Deque<TreeNode> nodeQueue = new LinkedList();
nodeQueue.add(root);
int i = 1;
while (!nodeQueue.isEmpty()) {
int len = nodeQueue.size();
List<Integer> cur = new ArrayList();
if (i > 0) {
while (len > 0) {
TreeNode node = nodeQueue.removeFirst();
cur.add(node.val);
if (node.left != null) {
nodeQueue.addLast(node.left);
}
if (node.right != null) {
nodeQueue.addLast(node.right);
}
len--;
}
} else {
while (len > 0) {
TreeNode node = nodeQueue.removeLast();
cur.add(node.val);
if (node.right != null) {
nodeQueue.addFirst(node.right);
}
if (node.left != null) {
nodeQueue.addFirst(node.left);
}
len--;
}
}
i *= -1;
res.add(cur);
}
return res;
}
}
一道中等难度题,再结合前两天的题目,相信以后碰到从上到下打印二叉树这题目,心里就有数了。
https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof