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

Posted 2021-12-13 谦谦均

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

例如：
给定二叉树 [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;
    }

解法不唯一，如果有错误，欢迎指正~