牛客网刷题-二叉树的之字形层序遍历

问题描述

给定一个二叉树，返回该二叉树的之字形层序遍历，（第一层从左向右，下一层从右向左，一直这样交替）

输入描述:
输入一棵树

输出描述:
输出树的之字形遍历

示例

示例1

输入
{3,9,20,#,#,15,7}

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

解决思路

分析

1. 本地很明显是一道广度优先搜索（bfs）的题目，广度优先搜索一般会借助队列这种数据结构，遍历每一层的节点，然后依次加入左节点和右节点，通过flag判断是否需要做结果的反转。

方法

1. 广度优先搜索

代码实现

// 思路1
public class Solution {
       
    public ArrayList<ArrayList<Integer>> zigzagLevelOrder(TreeNode root) {
     
        if (root == null) return new ArrayList<>();
        // write code here
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        boolean flag = true;
        while (!queue.isEmpty()) {
     
            ArrayList<Integer> list = new ArrayList<>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
     
                TreeNode node = queue.poll();
                list.add(node.val);
                if (node.left != null) queue.add(node.left);
                if (node.right != null) queue.add(node.right);
            }
            if (!flag) {
     
                Collections.reverse(list);
            }
            flag = !flag;
            result.add(list);
        }

        return result;
    }
}

时间复杂度分析：
O(M)：广度优先搜索需要遍历所有的节点。

空间复杂度分析：
O(1)：除了结果，没有使用额外的存储空间。

小伙伴如果想测试的话，可以直接到牛客网这个链接做测试

二叉树的之字形层序遍历-牛客网