循环不变量:以leetcode106. 从中序与后序遍历序列构造二叉树为例

给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。

示例 1:

输入:inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出:[3,9,20,null,null,15,7]

示例 2:

输入:inorder = [-1], postorder = [-1]
输出:[-1]

整体思路

题目根据两个顺序构造唯一的二叉树。那么构造二叉树的要点我们可以总结如下:

  • 找到中间节点
  • 切割两个数组

如果用两个序列可以很直观的看出来:

那么代码该如何写呢

说到一层一层切割,就应该想到了递归。

来看一下一共分几步:

     首先需要创建一个表来存数据和对应下标。

Map<Integer,Integer> map = new HashMap<>();
for(int i=0;i<inorder.length,i++){
        map.put(inorder[i],i);
}
  • 第一步:如果数组大小为零的话,说明是空节点了。

  • 第二步:如果不为空,那么取后序数组最后一个元素作为节点元素。

这里面是递归,那么依然是递归三步:

        1、 返回值和参数:

                        需要返回一个二维数组,需要不断切割数组,所以输入参数需要两个数组索引,数组对应的首位坐标。

public TreeNode build(int[] inorder,int inorderbegin,int inorderend,int[] postorder,int postorderbegin,int postorderend){
    return root;
}

        2、递归终止条件

// 对于数组来说,遍历的终止条件一般都是索引越界
// 注意这个“=”:看传入的数据的值,是中序和后序的最后一个值。
// 所以是左闭右闭区间

if(inorderbegin>=inoderend||postorderbegin>=postoderend){
        return null;
}

        3、一次递归逻辑

通过后序的最后一个值在表中找到对应中序所在的索引。这个逻辑在上面的图片中已经画出来了。

int rootIndex = map.get(postorder[postorder.length-1]);
TreeNode root = new TreeNode(inoder[rootIndex]);
// 根据rootIndex切分前后序----->左序长度
int lenOfLeft = rootIndex - inorderbegin;
root.left = build(inorder,inorderbegin,rootIndex,postorder,postoderbegin,postoderbegin+lenOfLeft);
root.right = 
// 左闭右开 
build(inorder,rootIndex + 1,inorderend,
postorder , lenOfLeft + postorderbegin , postorderend-1);
  • 第三步:找到后序数组最后一个元素在中序数组的位置,作为切割点

  • 第四步:切割中序数组,切成中序左数组和中序右数组 (顺序别搞反了,一定是先切中序数组)

  • 第五步:切割后序数组,切成后序左数组和后序右数组

  • 第六步:递归处理左区间和右区间

关键KEY

这个算法的关键是,在后序遍历序列中,根节点总是出现在序列的最后一个元素,因此在 postorder 数组中,我们知道根节点的位置是 postEnd - 1。而在中序遍历中,根节点将中序遍历序列分成了左子树和右子树,因此我们可以根据根节点在中序遍历中的位置 rootIndex 来确定左子树和右子树的范围。

为了构建右子树,我们需要跳过根节点,因此后序遍历的范围为 [postBegin + lenOfLeft, postEnd - 1]。这包括左闭和右闭,以确保根节点不被包含在右子树中。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值