leetcode103 二叉树的锯齿形层序遍历

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

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

3

/
9 20
/
15 7
返回锯齿形层序遍历如下:

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

链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
使用了双端队列存储,在进行层次遍历往队列里面加的时候仍然是按照层次遍历的顺序,但是当遍历读取的时候就要存储到一个双端队列里面,看看从哪一边进去

public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        if(root==null) return new ArrayList<>();
        LinkedList<TreeNode> queue=new LinkedList<>();
        List<List<Integer>> ans=new ArrayList<>();
        boolean flag=true;
        queue.addFirst(root);
        TreeNode cur=root;
        while (!queue.isEmpty())
        {
            LinkedList<Integer> sub=new LinkedList<>();
            int count=queue.size();
            for(int i=0;i<count;i++)
            {
                cur=queue.pollLast();
                if(flag)
                    sub.addLast(cur.val);
                else
                    sub.addFirst(cur.val);
                if(cur.left!=null) queue.addFirst(cur.left);
                if(cur.right!=null) queue.addFirst(cur.right);

            }
            ans.add(sub);
            flag=!flag;
        }
        return ans;
    }
posted @ 2021-09-02 11:35  LiangLiangAA  阅读(16)  评论(0编辑  收藏  举报
theme: { name: 'geek', avatar: '', headerBackground: '' // ... },