LeetCode 从中序遍历和后序遍历构建二叉树（java）

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

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

例如，给出

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

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

思路：

①后续遍历的最后一个数为根结点，根据这个根结点来划分中序遍历，将其分为左子树和右子树

②确定左右子树的中序遍历和后续遍历中的界限

③递归调用

 

代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        if(inorder.length == 0 || postorder.length == 0 || inorder.length != postorder.length){
            return null;
        }
        java.util.Map map = new java.util.HashMap<>();
        //将中序遍历的node和下标进行映射
        for(int i = 0;i < inorder.length;i++){
            map.put(inorder[i],i);
        }
        return buildNode(inorder,0,inorder.length-1,postorder,0,postorder.length-1,map);
    }
    
    public TreeNode buildNode(int[] inorder,int begin1,int end1, 
                              int[] postorder,int begin2,int end2,
                              java.util.Map map) {
         if(begin2 > end2) {
        	return null;
        }
        if(begin1 == end1){
            return new TreeNode(inorder[begin1]);
        }
        TreeNode root = new TreeNode(postorder[end2]);
        int index = map.get(postorder[end2]);
        root.left = buildNode(inorder,begin1,index-1,postorder,begin2,begin2+index-begin1-1,map);
        root.right = buildNode(inorder,index+1,end1,postorder,begin2+index-begin1,end2-1,map);
        return root;
    }
}