题目:
Given inorder and postorder traversal of a tree, construct the binary tree.
Note: You may assume that duplicates do not exist in the tree.
解答:
本题根据树的中序遍历和后序遍历,得到树的结构。与 LeetCode 105 思路基本一致
整体思路为:
后序遍历的最后一个节点一定是根节点,根据后序遍历确定根节点后,在中序中找到根节点的位置,则该位置之前的均为根节点的左子树,该位置之后的都是根节点的右子树,以此进行递归实现
具体代码实现如下:
- 用 post 指向后序遍历中,当前的根节点,初始指向后序遍历的最后一个节点
- 用 index 指向中序遍历中的根节点,则 index 左侧的为 root 的左子树,index 右侧的为 root 的右子树
- 递归中,inStart 和 inEnd 分别指向当前左子树/右子树的节点范围,分别递归,找到 root 的左子树和右子树
- 其中递归实现时,root 的右子树根节点位置指向后序遍历 postorder 中的 post-1,root 的左子树根节点位置指向后序遍历 postorder 中的 post-(inEnd-index+1),其中 inEnd-index+1 为根节点右子树的节点个数,则根节点的左子树节点指向 post 向前移动右子树节点的个数个位置
class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
if(inorder==null || postorder==null) {
return null;
}
return build(inorder, postorder, 0, inorder.length-1, postorder.length-1);
}
private TreeNode build(int[] inorder, int[] postorder, int inStart, int inEnd, int post) {
if(inStart > inEnd || post <0) {
return null;
}
TreeNode root = new TreeNode(postorder[post]);
int index = 0;
for(int i=inStart; i<=inEnd; i++) {
if(inorder[i] == root.val) {
index = i;
break;
}
}
root.right = build(inorder, postorder, index+1, inEnd, post-1);
root.left = build(inorder, postorder, inStart, index-1, post-(inEnd-index+1));
return root;
}
}