LeetCode 106 从中序与后序遍历序列构造二叉树（Java 版）

题目

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

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

示例：

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

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

解题思路

中序遍历：左 - 中 - 右
后序遍历：左 - 右 - 中

之前做过前序和中序建树，发现前序便于建树，中序用于分治法切割。

这里的话，不难发现从后面看后序遍历，它的顺序就是 中-右-左。和之前的前序中-左-右只有一点点区别，所以可以使用前序和中序建树的发自。 只不过需要将左右子树建树的顺序调转以下。

分治法

由于后序反向是 中右左。 所以应用分治法的时候，先建立右子树，再建立左子树

class Solution {
    int postIdx;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
    	// 去除不合理的情况（测试用例）
        if (inorder.length != postorder.length) return null;
        if (inorder.length == 0) return null;
        // 从后序遍历序列的结尾往前建树
        postIdx = postorder.length - 1;
        // 给定建树区间
        return buildTree(inorder, postorder, 0, postIdx);
    }
    public TreeNode buildTree(int[] inorder, int[] postorder, int start, int end){
    	// 没有建树的空间了，返回null
        if(start > end) return null;
        TreeNode node = new TreeNode(postorder[postIdx--]);
        // 是叶子节点，直接返回
        if(start == end) return node;
        // 找到当前元素在中序列表中的位置，以进行下一步的左子树、右子树建立
        int index = findIdx(inorder, node.val, end);
        // 由于是后序遍历序列倒着建树（中-右-左），因此先建立右子树
        node.right = buildTree(inorder, postorder, index+1, end);
        // 右子树建树
        node.left = buildTree(inorder, postorder, start, index-1);
        return node;
    }
    // 找到当前节点，在中序列表中的位置
    public int findIdx(int[] inorder, int val, int end){
        for(int i=end; i>=0; i--)
            if(inorder[i] == val) return i;
        return 0;
    }
}

Tip：注意自己先要测试一些测试用例（符合要求的、不符合要求的、不合理的输入）

图解

通过递归不断 左右子树 建树