力扣刷题：二叉树的锯齿形层次遍历（java实现）

原创于 2021-11-15 21:34:31 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #leetcode #算法

数据结构专栏收录该内容

27 篇文章

订阅专栏

本文详细讲解了如何使用双端队列和广度优先搜索策略，实现给定二叉树的锯齿形层序遍历。通过示例和代码演示，帮助读者理解如何根据层数奇偶性调整节点值的添加顺序。

题目：给定一个二叉树，返回其节点值的锯齿形层序遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

在这里插入图片描述
返回锯齿形层序遍历如下：

[
[3],
[20,9],
[15,7]
]

相关标签：树、广度优先搜索、二叉树

解析：题目要求按照层数的奇偶来存储数据的顺序，可以考虑双端队列，根据层数来将元素的值添加到队头或者队尾,，再用一个队列来维护当前层的结点。具体代码如下:

public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> lists = new LinkedList<>();
        if(root == null){
            return lists;
        }
        //创建一个队列用于存放每一层的结点
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);//初始的时候添加根结点
        boolean sign = true;//用于区分奇偶层
        while (!queue.isEmpty()){
            //获取队列的长度
            int size = queue.size();
            //创建一个双端队列用于存储该层的结点值
            Deque<Integer> res = new LinkedList<>();
            for (int i = 0; i < size; i++) {
                //获取队头结点
                TreeNode curNode = queue.poll();
                //根据层数来判断添加到队头还是队尾
                if(sign){
                    res.offerLast(curNode.val);
                }else {
                    res.offerFirst(curNode.val);
                }
                //更新结点队列
                if(curNode.left!=null){
                    queue.offer(curNode.left);
                }
                if(curNode.right!=null){
                    queue.offer(curNode.right);
                }
            }
            //添加到结果集里面
            lists.add(new LinkedList<>(res));
            //改变层数状态
            sign = !sign;
        }
        return lists;
    }