Python列表如何转化为⼆叉树?Day46: 列表转化为⼆叉树
已知列表nums,将其转化为⼆叉树。举例:
nums = [3,9,20,None,None,15,7],转化为⼆叉树后:
节点3的左⼦节点9,右⼦节点20,9的左右⼦节点都为None,20的左⼦节点15,右⼦节点7,参考下⾯:
⼆叉树定义:
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
请补全下⾯函数:
def list_to_binarytree(nums):
pass二叉树公式
构建分析
构建满⾜以上结构的⼆叉树,可以观察到:树的⽗节点和左右⼦节点的关系:
基于以上公式,再使⽤递归构建⼆叉树。
递归基情况:
if index >= len(nums) or nums[index] is None:
return None
递归⽅程:
根据以上得到如下代码:
代码
def list_to_binarytree(nums):
def level(index):
if index >= len(nums) or nums[index] is None:
return None
root = TreeNode(nums[index])
root.left = level(2 * index + 1)
root.right = level(2 * index + 2)
return root
return level(0)
binary_tree = list_to_binarytree([3,9,20,None,None,15,7])
以上递归版本,代码⽐较简洁。除此之外,为了训练算法思维,我们还可以使⽤迭代构建⼆叉树,使⽤队列数据结构。但是代码相对复杂⼀点,这个等再过⼏天布置这个作业。
《end》
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。
发表评论