已知一个二叉树的前序遍历序列和中序遍历序列,求其后序遍历序列。代码
时间: 2023-06-08 08:06:35 浏览: 109
二叉树问题
这是一道比较经典的二叉树遍历问题,可以通过递归的方式求解。具体思路如下:
1. 根据前序遍历序列确定根节点;
2. 在中序遍历序列中定位根节点位置,根节点左边的是其左子树的中序遍历序列,右边是其右子树的中序遍历序列;
3. 通过左子树和右子树的中序遍历序列长度,可以推算出前序遍历序列中左子树和右子树的范围,分别递归求解;
4. 最后将得到的左子树和右子树的后序遍历序列合并,再将根节点加入其中即可。
下面是示例代码:
```
public List<Integer> postorderTraversal(int[] preorder, int[] inorder) {
List<Integer> res = new ArrayList<>();
if(preorder == null || preorder.length == 0) {
return res;
}
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i < inorder.length; i++) {
map.put(inorder[i], i);
}
postorderHelper(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1, res, map);
return res;
}
private void postorderHelper(int[] preorder, int preStart, int preEnd, int[] inorder, int inStart, int inEnd, List<Integer> res, Map<Integer, Integer> map) {
if(preStart > preEnd || inStart > inEnd) {
return;
}
int rootVal = preorder[preStart];
int rootIndex = map.get(rootVal);
int leftLen = rootIndex - inStart;
int rightLen = inEnd - rootIndex;
postorderHelper(preorder, preStart + 1, preStart + leftLen, inorder, inStart, rootIndex - 1, res, map);
postorderHelper(preorder, preEnd - rightLen + 1, preEnd, inorder, rootIndex + 1, inEnd, res, map);
res.add(rootVal);
}
```
注意,在实际编写中,建议将 inorder 数组转换为 map 进行快速查找,避免每次都进行线性查找,提高算法性能。
阅读全文