2020/05/23 每日一题
经典的二叉树问题,需要重点理解好三种排序算法之间的区别。核心是利用好中序遍历作为标杆
前序遍历 [3 9 15 20 7]
中序遍历 [9 3 15 20 7]
后序遍历 [9 15 7 20 3]
最特殊的就是中序遍历算法,每一个根节点的两侧的数据都是两棵子树的数据。因此可以作为标杆使用。
我们结合题目来分析。
迭代方法
主要分析迭代的方法。
步骤:
- 放入根节点,这里是前序遍历的第一个数字;维护一个index,这里是从0开始,index对应中序遍历。
- 开始根据前序遍历,当
stack[-1].val != inorder[index]
是,依次构建每个节点的左子树,并将左子树加入栈。 - 如果不满足上诉条件,开始弹栈并且index+1,目标是弹到栈顶的元素与inorder[index]不同。
- 对最后弹出的节点node建立右子树,并将右子树加入栈。
class Solution(object):
def buildTree(self, preorder, inorder):
## 采用迭代
if not preorder:
return None
root = TreeNode(preorder[0])
index = 0
stack = [root]
for i in range(1,len(preorder)):
node = stack[-1]
if node.val != inorder[index]:
node.left