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

链接

LeetCode：106. 从中序与后序遍历序列构造二叉树

思路

同 《剑指offer》系列 重建二叉树（Java） 一样，后序遍历的最一个值为根节点的值，使用这个值将中序遍历结果分成两部分，左部分为树的左子树中序遍历的结果，右部分为树的右子树中序遍历的结果。

代码

class Solution {
     
    public TreeNode buildTree(int[] inorder, int[] postorder) {
     
        int n = postorder.length;
        return CreateByMidPost(inorder, postorder, 0, n-1, n);
    }

    /** index_in是inorder数组的下标 index_post是postorder数组的下标
     *   n是二叉树的大小
     *   后序遍历的尾节点就是根节点 所以postorder[index_post]就是树根节点的值、
     *   再找到中序遍历中对应的值 该值左边的值就是左子树对应的 右边的值就是右子树对应的
     *   左子树 右子树的情况同样如此
     */
    public TreeNode CreateByMidPost(int[] inorder, int[] postorder, int index_in, int index_post, int n) {
     
        if (n == 0)
            return null;
        TreeNode p = new TreeNode(postorder[index_post]);
        int i;                          //这里的i相当于左子树的个数
        for (i = 0; i < n; i++) {
     
            if (inorder[index_in + i] == postorder[index_post])
                break;
        }
        p.left = CreateByMidPost(inorder, postorder, index_in, index_post - n + i, i);
        p.right = CreateByMidPost(inorder, postorder, index_in + i + 1, index_post - 1, n - i - 1);
        return p;
    }

}