200字范文,内容丰富有趣,生活中的好帮手!
200字范文 > 二叉树高度的代码解析_剑指offer 从上到下打印二叉树

二叉树高度的代码解析_剑指offer 从上到下打印二叉树

时间:2021-07-15 02:04:12

相关推荐

二叉树高度的代码解析_剑指offer 从上到下打印二叉树

题目描述

从上往下打印出二叉树的每个节点,同层节点从左至右打印。

解析:这实际上就是二叉树的层序遍历。规律如下:每次打印节点,如果该节点有子节点,则将子节点放入队列末尾,然后取出队列头部最早进入队列的节点,重复上述操作直至队列为空。用list模拟队列。

代码如下:

class Solution:# 返回从上到下每个节点值列表,例:[1,2,3]def PrintFromTopToBottom(self, root):# write code hereres = []queue = []if not root:return resqueue.append(root)while queue:tmp = queue.pop(0)res.append(tmp.val)if tmp.left:queue.append(tmp.left)if tmp.right:queue.append(tmp.right)return res

扩展1:分行从上到下打印二叉树

题目描述

从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

解析:层序遍历的变种,本质相同,方法类似。

代码如下:

class Solution:# 返回二维列表[[1,2],[4,5]]def Print(self, pRoot):# write code hereres = []if not pRoot:return resqueue = [pRoot]while queue:level = []next = []for tmp in queue:level.append(tmp.val)if tmp.left:next.append(tmp.left)if tmp.right:next.append(tmp.right)res.append(level)queue = next[:]return res

扩展2:之字形打印二叉树

题目描述

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

解析:层序遍历的变种。

代码如下:

class Solution:def Print(self, pRoot):# write code hereres = []if not pRoot:return resqueue = [pRoot]k = 0while queue:level = []next = []for tmp in queue:level.append(tmp.val)if tmp.left:next.append(tmp.left)if tmp.right:next.append(tmp.right)if k & 1 == 0:res.append(level)else:level.reverse()res.append(level)queue = next[:]k = k + 1return res

.3.21

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。