Leetcode 106.从中序与后序遍历序列构造二叉树

Posted 2021-01-29 kexinxin

从中序与后序遍历序列构造二叉树

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

中序遍历 inorder = [9,3,15,20,7]

后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树：

3

/

9 20

/

15 7

 

解题思路：

 

已知中序遍历和后序遍历IN和Post，求还原二叉树。

 

后序遍历的最后一个数post[len-1]就是root节点。

 

搜索IN，如果IN[I]=post[len-1]，那么 IN[I+1]-IN[len-1]就是root的右子树，IN[0]-IN[I-1]就是root的左子树。

 

通过递归求解。

 

 1 public class Solution{
 2     public TreeNode buildTree(int[] inorder,int[] postorder){
 3         return build(postorder.length-1,0,inorder.length-1,inorder,postorder);
 4     }
 5 
 6     private TreeNode build(int poststart,int instart,int inend,int[] inorder,int[] postorder){
 7         if(poststart<0 || instart>inend){
 8             return null;
 9         }
10         TreeNode root=new TreeNode(postorder[poststart]);
11         int inindex=0;
12         for(int i=instart;i<=inend;i++){
13             if(inorder[i]==root.val){
14                 inindex=i;
15                 break;
16             }
17         }
18         root.right=build(poststart-1,inindex+1,inend,inorder,postorder);
19         root.left=build(poststart-(inend-inindex)-1,instart,inindex-1,inorder,postorder);
20         return root;
21     }
22 }