一、题目
请实现一个函数按照之字形顺序打印二叉树,即:第一行
按照从左到右的顺序打印,第二层
按照从右到左的顺序打印,第三行
再按照从左到右的顺序打印,其他行以此类推。
二、示例
2.1> 示例1
提示:
- 节点总数 <=
1000
三、解题思路
本题是算法《剑指 Offer 32 - II. 从上到下打印二叉树 II》题的变形,在原有层序遍历的基础上,根据奇数层按照由左到右
进行输出,而根据偶数层则按照从右向左
进行输出;
那么层序遍历我们之前的题解中提到过,通过采用双向队列Deque<TreeNode> deque以及配合当前层级的节点数num就可以实现按层遍历操作,那么本题的难点就在于如何根据奇数/偶数层数来转换遍历节点。
为了实现遍历结果的反转,我们可以再创建一个变量LinkedList<Integer> item,由于LinkedList提供了从头部开始链接的addFirst(...)
方法和从尾部开始链接的addLast(...)
方法,所以我们只需执行如下操作:
【奇数层】调用
addLast(...)
方法进行item的子结果拼装;
【偶数层】调用addFirst(...)
方法进行item的子结果拼装;
那么最终将每层的item组合到最终结果List<List<Integer>> result即可。根据上面介绍的解题思路,我们以二叉树结构[1,2,3,4,5,6,7]
为例,具体看一下针对这道题的处理逻辑。请见下图所示:
四、代码实现
public class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList();
if (root == null) return result;
Deque<TreeNode> deque = new ArrayDeque();
deque.addLast(root);
int num;
boolean reverse = false;
while(!deque.isEmpty()) {
num = deque.size();
LinkedList<Integer> item = new LinkedList<>();
for (int i = 0; i < num; i++) {
TreeNode node = deque.removeFirst();
if (reverse) item.addFirst(node.val);
else item.addLast(node.val);
if (node.left != null) deque.addLast(node.left);
if (node.right != null) deque.addLast(node.right);
}
result.add(item);
reverse = !reverse;
}
return result;
}
}
今天的文章内容就这些了:
写作不易,笔者几个小时甚至数天完成的一篇文章,只愿换来您几秒钟的 点赞 & 分享 。
更多技术干货,欢迎大家关注公众号“爪哇缪斯” ~ \(o)/ ~ 「干货分享,每天更新」
网友评论