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

这是树的第10篇算法,力扣链接

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

示例 1:

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

回忆一下中序遍历和后序遍历。

中序遍历 (Inorder Traversal)

在中序遍历中,我们按照以下顺序遍历树中的节点:

  1. 遍历左子树
  2. 访问根节点
  3. 遍历右子树

对于二叉搜索树(BST),中序遍历会按照升序访问所有节点,因为二叉搜索树的特点是左子节点的值小于根节点的值,根节点的值小于右子节点的值。

后序遍历 (Postorder Traversal)

后序遍历的顺序是:

  1. 遍历左子树
  2. 遍历右子树
  3. 访问根节点

后序遍历常用于删除或释放树中的节点。因为你在删除节点之前先访问其子节点,这样可以安全地删除每个节点。

假设有一棵二叉树如下:

    A
   / \
  B   C
 / \   \
D   E   F

对这棵树进行不同的遍历会得到以下结果:

  • 中序遍历D, B, E, A, C, F。首先遍历左子树(D, B, E),然后是根节点(A),最后是右子树(C, F)。
  • 后序遍历D, E, B, F, C, A。首先是左子树(D, E, B),然后是右子树(F, C),最后是根节点(A)。

这道题和上一篇算法的前序遍历类似,这不过这里的后序遍历涉及到一定的逆逻辑,例如根节点是最后一个节点,右子树的范围到 len(postorder)-1-len(inorder[index+1:])

func buildTree(inorder []int, postorder []int) *TreeNode {
	if len(postorder) == 0 {
		return nil
	}
	index := 0
	for ; index < len(inorder); index++ {
		if inorder[index] == postorder[len(postorder)-1] {
			break
		}
	}
	result := &TreeNode{Val: postorder[len(postorder)-1]}
	result.Right = buildTree(inorder[index+1:], postorder[len(postorder)-1-len(inorder[index+1:]):len(postorder)-1])
	result.Left = buildTree(inorder[:index], postorder[:len(postorder)-1-len(inorder[index+1:])])
	return result
}

当然,也可以试试迭代的方法。

func buildTree(inorder []int, postorder []int) *TreeNode {
	if len(postorder) == 0 {
		return nil
	}
	result := &TreeNode{Val: postorder[len(postorder)-1]}
	stack := []*TreeNode{result}
	inorderIndex := len(inorder) - 1
	for i := len(postorder) - 2; i >= 0; i-- {
		node := stack[len(stack)-1]
		if node.Val != inorder[inorderIndex] {
			node.Right = &TreeNode{Val: postorder[i]}
			stack = append(stack, node.Right)
		} else {
			for len(stack) > 0 && stack[len(stack)-1].Val == inorder[inorderIndex] {
				node = stack[len(stack)-1]
				stack = stack[:len(stack)-1]
				inorderIndex--
			}
			node.Left = &TreeNode{Val: postorder[i]}
			stack = append(stack, node.Left)
		}
	}
	return result
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值