【LeetCode面试经典150题】106. 从中序和后序遍历构造二叉树

最新推荐文章于 2024-07-27 17:45:51 发布

奕_Raku

最新推荐文章于 2024-07-27 17:45:51 发布

阅读量363

点赞数 3

分类专栏： LeetCode 文章标签： leetcode 面试算法

本文链接：https://blog.csdn.net/qq_29536773/article/details/139849155

版权

LeetCode 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

一、题目

106. 从中序与后序遍历序列构造二叉树 - 力扣（LeetCode）
给定两个整数数组 inorder 和 postorder ，其中 inorder 是二叉树的中序遍历， postorder 是同一棵树的后序遍历，请你构造并返回这颗 二叉树 。

二、思路

思路和前&中序遍历一致，关键点在于如何划分各种排序的左子树部分和右子树部分并送入下一层递归，见解法一；

三、解法

解法一

class Solution {
    private final HashMap<Integer, Integer> map = new HashMap();

    public TreeNode buildTree(int[] inorder, int[] postorder) {
        int inB = 0, inE = inorder.length,
                postB = 0, postE = postorder.length;
        for (int i = 0; i < inorder.length; i++) {
            map.put(inorder[i], i);
        }
        return recursion(inorder, inB, inE, postorder, postB, postE);
    }

    private TreeNode recursion(
            int[] inorder, int inB, int inE,
            int[] postorder, int postB, int postE) {
        // 递归终止条件
        if (inB >= inE || postB >= postE) {
            return null;
        }
        // 找到根结点
        int rootVal = postorder[postE - 1];
        int rootIndexInIn = map.get(rootVal);
        // 连接当前结点
        TreeNode node = new TreeNode(rootVal);
        // 左子树
        int leftLen = rootIndexInIn - inB;
        node.left = recursion(
            /* 9 (3) 15 20 7 */
            inorder, inB, rootIndexInIn, 
            /* 9 15 7 20 (3) */
            postorder, postB, postB + leftLen
        );
        // 右子树
        node.right = recursion(
            /* 9 (3) 15 20 7 */
            inorder, rootIndexInIn + 1, inE, 
            /* 9 15 7 20 (3) */
            postorder, postB + leftLen, postE - 1
        );
        return node;
    }
}