Problem:
Given inorder and postorder traversal of a tree, construct the binary tree.
Explanation:
将中序遍历和后序遍历还原成二叉树
My Thinking:
后序遍历的最后一个节点一定是根结点,在inorder中找到它的位置,分为左右两半构成左右子树,左孩子一定是postorder当前-右子树的个数-1,右孩子一定是postorder中的前一个值。
My Solution:
class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
return build(postorder.length-1,0,inorder.length-1,inorder,postorder);
}
public TreeNode build(int curpost,int instart,int inend,int[] inorder, int[] postorder){
if(curpost<0 || instart>inend)
return null;
TreeNode root=new TreeNode(postorder[curpost]);
//找到根结点在inorder中的位置
int index=0;
for(int i=0;i<inorder.length;i++){
if(inorder[i]==postorder[curpost]){
index=i;
break;
}
}
//左孩子一定是postorder当前-右子树的个数-1
root.left=build(curpost-(inend-index)-1,instart,index-1,inorder,postorder);
//右孩子一定是postorder中的前一个值
root.right=build(curpost-1,index+1,inend,inorder,postorder);
return root;
}
}
Optimum Thinking:
Optimum Solution: