BFS刷题笔记_二叉树齿轮层序遍历

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练练手

[二叉树齿轮层序遍历]