给定二叉树的后序遍历和中序遍历,输出前序遍历
时间: 2023-12-14 12:31:44 浏览: 138
根据给定的后序遍历和中序遍历,可以构造出二叉树,然后再进行前序遍历输出。具体步骤如下:
1.定义一个TreeNode类,包含val、left、right三个属性,用于表示二叉树的节点。
2.定义一个函数buildTree,接收两个参数inorder和postorder,分别表示中序遍历和后序遍历。
3.在buildTree函数中,首先判断inorder和postorder是否为空,如果为空,则返回None。
4.然后从postorder中取出最后一个元素,作为当前子树的根节点。
5.在inorder中找到根节点的位置,将inorder分成左子树和右子树两部分。
6.递归调用buildTree函数,分别传入左子树的中序遍历和后序遍历,以及右子树的中序遍历和后序遍历,得到左子树和右子树的根节点。
7.将左子树和右子树的根节点分别作为当前根节点的左右子节点。
8.最后返回当前根节点。
9.定义一个函数preorderTraversal,接收一个参数root,表示二叉树的根节点。
10.在preorderTraversal函数中,首先判断root是否为空,如果为空,则返回空列表。
11.然后按照根节点、左子树、右子树的顺序进行前序遍历,将遍历结果存入列表中。
12.最后返回列表。
下面是完整的代码实现:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def buildTree(inorder, postorder):
if not inorder or not postorder:
return None
root_val = postorder[-1]
root = TreeNode(root_val)
index = inorder.index(root_val)
left_inorder = inorder[:index]
right_inorder = inorder[index+1:]
left_postorder = postorder[:index]
right_postorder = postorder[index:-1]
root.left = buildTree(left_inorder, left_postorder)
root.right = buildTree(right_inorder, right_postorder)
return root
def preorderTraversal(root):
if not root:
return []
res = [root.val]
res += preorderTraversal(root.left)
res += preorderTraversal(root.right)
return res
inorder = [9,3,15,20,7]
postorder = [9,15,7,20,3]
root = buildTree(inorder, postorder)
print(preorderTraversal(root)) # 输出:[3, 9, 20, 15, 7]
```
阅读全文