LeetCode【103】 二叉树的锯齿形层次遍历

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

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

3

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

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

注:和102几乎没有区别,添加一个line变量,根据奇数行还是偶数行,决定正插还是反插。

public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> resultList = new ArrayList<>();
        if(root == null)
            return resultList;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);

        int line = 0;
        //将每层的节点放入Queue
        while(!queue.isEmpty()) {
            LinkedList<Integer> levelList = new LinkedList<>();
            //用count来限制每层poll几个节点
            int count = queue.size();
            for(int i=0 ;i<count;i++){
                TreeNode now = queue.poll();
                if(line%2==0){
                    levelList.add(now.val);
                }else{
                    levelList.addFirst(now.val);
                }
                if(now.left!=null){
                    queue.add(now.left);
                }
                if(now.right!=null){
                    queue.add(now.right);
                }
            }
            resultList.add(levelList);
            line++;
        }
        return resultList;
    }
发布了55 篇原创文章 · 获赞 14 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq422243639/article/details/103752530
今日推荐