其他分享
首页 > 其他分享> > 牛客网刷题-二叉树的之字形层序遍历

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

作者:互联网

问题描述

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

输入描述:
输入一棵树

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

示例

示例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):除了结果,没有使用额外的存储空间。

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

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

标签:node,之字形,ArrayList,层序,flag,queue,add,遍历,二叉树
来源: https://blog.csdn.net/qq_35398517/article/details/113768701