1.题目描述:
给你二叉树的根节点 root
,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
示例1:
示例 2:
输入:root = [1] 输出:[[1]]
示例 3:
输入:root = [] 输出:[]
提示:
-
树中节点数目在范围
[0, 2000]
内 -
-100 <= Node.val <= 100
-
二叉树的结构如下:
-
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode() {}
TreeNode(int val) { this.val = val; }
TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
思路:
二叉树的齿轮层序遍历, 其实是BFS 算法,二叉树层序遍历的变形题,我们不妨在二叉树层序遍历的基础上做修改,可以在二叉树层序遍历算法的基础上,使用两端队列,实现从左往右或者从右往左的遍历,进入待选队列时还是正常队列进去,在两端队列时, addlast()则是从左往右,addfirst()则是从右往左;
解法:
1.把root加入待选队列nodeQueue,定义ltr = true 表示从左到右
2.当nodeQueue队列不为空时,循环如下步骤
(1)建一个两端队列dqueue,取nodeQueue得size
(2)循环从nodeQueue队列中取size个元素,为当前层,每次取出的元素定义为currnode
1)如果当前层是从左到右,则dqueue进行addlast(currnode.val)否则 addfirst(currnode.val)
2)如果当前节点的左子节点不为空,对左子节点进行扩展
3)如果当前节点的右子节点不为空,对右子节点进行扩展
(3)将当前层dqueue加入ans
(4)调转ltr信号
java实现代码如下:
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> ans = new LinkedList<>();
boolean ltr = true;
if (root == null) {
return ans;
}
Queue<TreeNode> nodeQueue = new LinkedList<>();
nodeQueue.offer(root);
while (!nodeQueue.isEmpty()) {
Deque<Integer> deque = new LinkedList<>();
int size = nodeQueue.size();
for (int i = 0; i < size; i++) {
TreeNode currNode = nodeQueue.poll();
if (ltr) {
//每次加在队列尾端就能保证从左往右
deque.addLast(currNode.val);
} else {
//每次加在队列前端,就能保证从右往左
deque.addFirst(currNode.val);
}
if (currNode.left != null) {
nodeQueue.offer(currNode.left);
}
if (currNode.right != null) {
nodeQueue.offer(currNode.right);
}
}
ans.add(new LinkedList<>(deque));
ltr = !ltr;
}
return ans;
}
如果理解了,不妨去leetcode练练手