力扣刷题:二叉树的锯齿形层次遍历(java实现)

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

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


image.png

返回锯齿形层序遍历如下:

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

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

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

public List> zigzagLevelOrder(TreeNode root) {
        List> lists = new LinkedList<>();
        if(root == null){
            return lists;
        }
        //创建一个队列用于存放每一层的结点
        Queue queue = new LinkedList<>();
        queue.offer(root);//初始的时候添加根结点
        boolean sign = true;//用于区分奇偶层
        while (!queue.isEmpty()){
            //获取队列的长度
            int size = queue.size();
            //创建一个双端队列用于存储该层的结点值
            Deque 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;
    }

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

你可能感兴趣的:(力扣刷题:二叉树的锯齿形层次遍历(java实现))