牛客网刷题-二叉树的之字形层序遍历
作者:互联网
问题描述
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
输入描述:
输入一棵树
输出描述:
输出树的之字形遍历
示例
示例1
输入
{3,9,20,#,#,15,7}
输出
[
[3],
[20,9],
[15,7]
]
解决思路
分析
- 本地很明显是一道广度优先搜索(bfs)的题目,广度优先搜索一般会借助队列这种数据结构,遍历每一层的节点,然后依次加入左节点和右节点,通过flag判断是否需要做结果的反转。
方法
- 广度优先搜索
代码实现
// 思路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