从中序与后序遍历序列构造二叉树

力扣地址:
https://leetcode.cn/problems/construct-binary-tree-from-inorder-and-postorder-traversal/

根据一棵树的中序遍历与后序遍历构造二叉树。
注意: 你可以假设树中没有重复的元素。
例如,给出
● 中序遍历 inorder = [9,3,15,20,7]
● 后序遍历 postorder = [9,15,7,20,3] 返回如下的二叉树:
tree

思考过程:
1、根据后序结果,可以确到root节点的位置,即后序的最后一个元素
2、找到了root节点,去中序就可以确认左子树和右子树
3、确认递归退出条件,当不满足左闭右开,说明没有元素,返回空树

class Solution {
    Map<Integer, Integer> map;  // 方便根据数值查找位置
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        map = new HashMap<>();
        // 用map保存中序序列的数值对应位置
        for (int i = 0; i < inorder.length; i++) {
            map.put(inorder[i], i);
        }

        return buildNode(inorder,  0, inorder.length, postorder,0, postorder.length);  // 前闭后开
    }

    public TreeNode buildNode(int[] inorder, int inBegin, int inEnd, int[] postorder, int postBegin, int postEnd) {
        // 参数里的范围都是前闭后开
        // 不满足左闭右开,说明没有元素,返回空树
        if (inBegin >= inEnd || postBegin >= postEnd) {
            return null;
        }
        // 找到后序遍历的最后一个元素在中序遍历中的位置
        int rootIndex = map.get(postorder[postEnd - 1]);
        // 构造结点
        TreeNode root = new TreeNode(inorder[rootIndex]);
        // 保存中序左子树个数,用来确定后序数列的个数
        int lenOfLeft = rootIndex - inBegin;
        // 注意,位置都是左闭右开的
        // 构建左子树,中序的开始位置不变,结束位置是后续的最后一个节点在中序的位置
        // 再看后序,后续的开始位置不变,后续的结束位置是后续从头数,左子树节点的个数
        // 左子树节点的个数是从中序中计算出来的
        root.left = buildNode(inorder, inBegin, rootIndex,
                            postorder, postBegin, postBegin + lenOfLeft);
        // 构建右子树,中序的开始位置,即root节点位置+1,结束位置就是中序的结束位置不变
        // 后续的开始位置是后续从头数,左子树节点的个数,结束位置是后序剔除root节点后的长度
        // 也就是后续的结束位置减一
        root.right = buildNode(inorder, rootIndex + 1, inEnd,
                            postorder, postBegin + lenOfLeft, postEnd - 1);

        return root;
    }
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值