温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

如何进行python二叉树的层次遍历

发布时间:2021-12-13 15:15:33 阅读:267 作者:柒染 栏目:大数据
Python开发者专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

本篇文章给大家分享的是有关如何进行python二叉树的层次遍历,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。

给定一个二叉树,返回其按层次遍历的节点值。(即逐层地,从左到右访问所有节点)。

例如:

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

    3

   / \

  9  20

    /  \

   15   7

返回其层次遍历结果:

[

  [3],

  [9,20],

  [15,7]

]

答案:

 1public List<List<Integer>> levelOrder1(TreeNode root) { 2    Queue<TreeNode> queue = new LinkedList<TreeNode>(); 3    List<List<Integer>> wrapList = new LinkedList<List<Integer>>(); 4    if (root == null) 5        return wrapList; 6    queue.offer(root); 7    while (!queue.isEmpty()) { 8        int levelNum = queue.size(); 9        List<Integer> subList = new LinkedList<Integer>();10        for (int i = 0; i < levelNum; i++) {11            if (queue.peek().left != null)12                queue.offer(queue.peek().left);13            if (queue.peek().right != null)14                queue.offer(queue.peek().right);15            subList.add(queue.poll().val);16        }17        wrapList.add(subList);18    }19    return wrapList;20}

解析:

LinkedList是个链表,先进先出,levelNum是每一层的节点数量,一层一层的遍历,代码没什么难度,下面再来看一种递归的写法

 1public List<List<Integer>> levelOrder(TreeNode root) { 2    List<List<Integer>> res = new ArrayList<List<Integer>>(); 3    levelHelper(res, root, 0); 4    return res; 5} 6 7public void levelHelper(List<List<Integer>> res, TreeNode root, int height) { 8    if (root == nullreturn; 9    if (height >= res.size()) {10        res.add(new LinkedList<Integer>());11    }12    res.get(height).add(root.val);13    levelHelper(res, root.left, height + 1);14    levelHelper(res, root.right, height + 1);15}

height其实相当于层级,height>=res.size()说明已经到了下一层了,所以要新建一个list。

以上就是如何进行python二叉树的层次遍历,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注亿速云行业资讯频道。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

原文链接:https://my.oschina.net/u/1010616/blog/4440476

AI

开发者交流群×