Construct Binary Tree from Inorder and Postorder Traversal

Construct Binary Tree from Inorder and Postorder Traversal

  Total Accepted: 3984  Total Submissions: 15858 My Submissions

Given inorder and postorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.

    3
   / \
  9  20
    /  \
   15   7

 

inorder = {9, 3, 15, 20, 7}

postorder = {9, 15, 7, 20, 3}

从inorder 和 postorder 中，可以发现规律。

post 的最后一个总是该subtree 的 root, 同时，在inorder中搜索 postorder的最后一个元素的 index, 这个index 就把inorder 二分成左subtree 和右subtree. 例如 postorder 最后的3，在inorder中把整个tree 二分成 {9} 和 {15, 20, 7} 左右两个subtree.

所以可以用递归来解，因为要搜索postorder最后元素在inorder中的index, 可以事先做一个map.节约时间。

注意：最后在写递归的时候，postEnd 特别要注意, 对于左subtree, 他是postEnd-1 - (inEnd-rootIndex) 。 就是说，postEnd-1 是剔除当前的postEnd, 同时还要减去右subtree所占用的length. 因为postorder是左在前右在后。

    public static TreeNode buildTree(int[] inorder, int[] postorder){
        Map<Integer, Integer> map = new HashMap<>();
        for(int i = 0; i < inorder.length; i++){
            map.put(inorder[i],i);
        }
        return buildTreeHelper(inorder,postorder,0,inorder.length-1,postorder.length-1,map);
    }

    private static TreeNode buildTreeHelper(int[] inorder, int[] postorder, int start, int end, int postEnd, Map<Integer, Integer> map){
        if(start > end) return null;
        int rootIndex = map.get(postorder[postEnd]);
        TreeNode root = new TreeNode(postorder[postEnd]);
        root.left = buildTreeHelper(inorder,postorder,start,rootIndex-1,postEnd -1 -(end-rootIndex), map);
        root.right = buildTreeHelper(inorder,postorder,rootIndex+1,end,postEnd -1, map);
        return root;
    }