leetcode刷题题解——106. 从中序与后序遍历序列构造二叉树

private Map<Integer,Integer> indexMap;
private int postindex;

public TreeNode buildTree(int[] inorder, int[] postorder) {
    indexMap = new HashMap<>();
    int len = postorder.length;
    postindex = len - 1;
    for (int i=0;i<len;i++) indexMap.put(inorder[i],i);

    return buildTree(inorder,postorder,0,len-1);
}

private TreeNode buildTree(int[] in,int[] post,int in_left,int in_right){
    if (in_left>in_right) return null;
    TreeNode root = new TreeNode(post[postindex--]);
    int index = indexMap.get(root.val);
    root.right = buildTree(in,post,index+1,in_right);
    root.left = buildTree(in,post,in_left,index-1);
    return root;
}

思路：
首先我们要明白，
中序遍历
[(左子树),根节点,(右节点)]
后序遍历
[(左子树),(右子树),根节点]

• 主函数部分
  代码中的Map，则是用于将inorder的所有值与其对应索引绑定，到需要利用前序遍历中获取的根节点来获取根节点在inorder中的索引时，时间复杂度能够为O(1)
• 建树函数
  1. 当左索引大于右索引时，说明该区间已经没有需要创建的结点，直接返回null即可
  2. 当前后序遍历的区间尾元素为树的根节点的值，创建一个int变量保存在后序遍历中根节点的索引
  3. 后序遍历中根节点的索引获取根节点的值，同时进行索引自减，再根据该值创建根节点，并利用Map获取其在inorder中的索引
  4. 利用index递归建立左右子树

     这里要和利用前序和中序遍历构建二叉树区别开来，这里伴随着postindex的自增，最先遇到的是右子树的根节点，所以先递归创建的应是root.right节点

  5. 返回root
• 建树完毕