根据二叉树的先序和中序数组直接生成后续数组

# 思路,根据先序遍历中,第一个数字为根,
# 根据根节点数字在中序遍历中找到其index,index左边的都是左子树,右边的都是右子树,记录左子树个数
# 先序遍历的第一个根,就是在后序遍历最后一个位置,根据算出的左子树个数,划分先序和后续数组的左右子树index
# 递归
def findorder(preorder, inorder):
    presize = len(preorder)
    res = [0] * presize
    findordersub(preorder, 0, presize - 1, inorder, 0, presize - 1, res, 0, presize - 1)
    return res


def findordersub(preorder, pl, pr, inorder, il, ir, res, rl, rr):
    if pl > pr:
        return
    root = preorder[pl]
    res[rr] = root
    idx = 0
    for idx in range(il, ir+1):
        if inorder[idx] == root:
            break
        else:
            idx += 1

    leftcount = idx - il

    # 注意每个数组的开始和结束index,可以写个例子对比
    findordersub(preorder, pl + 1, pl + leftcount, inorder, il, idx - 1, res, rl, rl + leftcount - 1)
    findordersub(preorder, pl + leftcount + 1, pr, inorder, idx + 1, ir, res, rl + leftcount, rr - 1)


print(findorder([6, 4, 2, 5, 3, 1, 7], [4, 2, 5, 6, 1, 3, 7]))

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值