刷题系列 - 中序和后序遍历队列,构造对应二叉树;
假期继续刷题,也没有别的什么事情可以干。
这个题是给出中序和后序遍历队列,构造对应二叉树;题目很简单,如下图,给出两个遍历队列,构成二叉树,这里假定没有重复点。
想了好几天,真是惭愧,因为一直想一次遍历就完成构造,最后发现不行;然后就硬搞出一个多重循环的遍历方法,虽然可行,但是提交后提示耗时超过限制。最后还是用递归实现的。
其实原理很简单,对于后续遍历队列,最后一个值就是整个二叉树的根节点;而这个根节点去掉后,可以把二叉树分成左右两个树,在中序队列中,按照这个根节点来拆分出可以得到左右队列,分布对应左边树和右边树的所有点。而且其实后序队列也是按照左右树节点划分的,只要知道左右树的节点数量,来划分就可以了,这个可以从中序队列划分结果获得。反复同理,再划分出来左右树继续划分,可以得到叶子节点,或者空序列;这样就完成树的构成。
代码如下:
#Definitionforabinarytreenode.#classTreeNode:#def__init__(self,x):#self.val=x#self.left=None#self.right=NoneclassSolution:defbuildTree(self,inorder:List[int],postorder:List[int])->TreeNode:ifinorder==[]:returnNoneelse:iflen(inorder)==1:returnTreeNode(inorder[0])else:RootVal=postorder[-1]currentNode=TreeNode(RootVal)inorderLeft=inorder[:inorder.index(RootVal)]inorderRight=inorder[inorder.index(RootVal)+1:]postorder.pop()postorderLeft=postorder[:len(inorderLeft)]postorderRight=postorder[-len(inorderRight):]currentNode.left=self.buildTree(inorderLeft,postorderLeft)currentNode.right=self.buildTree(inorderRight,postorderRight)returncurrentNode
声明:本站所有文章资源内容,如无特殊说明或标注,均为采集网络资源。如若本站内容侵犯了原著者的合法权益,可联系本站删除。