LeetCode:Construct Binary Tree from Inorder and Postorder Traversal

Construct Binary Tree from Inorder and Postorder Traversal




Total Accepted: 57352  Total Submissions: 195355  Difficulty: Medium

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

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

Subscribe to see which companies asked this question

Hide Tags
  Tree Array Depth-first Search
Hide Similar Problems
  (M) Construct Binary Tree from Preorder and Inorder Traversal
















思路:

根据“后序”数组的最后一个结点,在“中序”数组中找到“根”结点index;根据index把“中序”分成index的左右子树。


java code:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        
        if(inorder == null || postorder == null || inorder.length != postorder.length) return null;
        
        HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
        for(int i=0;i<inorder.length;i++)
            map.put(inorder[i], i);
        
        return buildTree(inorder, 0, inorder.length-1, postorder, 0, postorder.length-1, map);
    }
    
    // 自定义函数
    private TreeNode buildTree(int[] inorder,int s1, int e1, int[] postorder,int s2, int e2, Map<Integer, Integer> map) {
        
        if(s1 > e1 || s2 > e2) return null;
        TreeNode root = new TreeNode(postorder[e2]);
        
        int rootIndex = map.get(postorder[e2]); // 找到中序遍历的“根”结点
        root.left = buildTree(inorder, s1, rootIndex-1, postorder, s2, s2+rootIndex-s1-1, map);
        root.right = buildTree(inorder, rootIndex+1, e1, postorder, s2+rootIndex-s1, e2-1, map);
        return root;
    }
}


你可能感兴趣的:(LeetCode,array,tree,search,Depth-first)