二叉树的中序遍历_Go 刷 leetcode从前序与中序遍历序列构造二叉树

最新推荐文章于 2023-01-18 15:06:54 发布

weixin_39603505

最新推荐文章于 2023-01-18 15:06:54 发布

阅读量88

点赞数

文章标签：二叉树的中序遍历二叉树遍历序列还原排序二叉树的中序遍历序列是有序序列由双遍历序列构造二叉树

今天为大家讲解 LeetCode 第 105 题，是一道关于数组和树的题目。

题目描述

根据一棵树的前序遍历与中序遍历构造二叉树。

注意: 你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7] 中序遍历 inorder = [9,3,15,20,7] 返回如下的二叉树：
    3
   / \
  9  20
    /  \
   15   7
来源：力扣(LeetCode)
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

preorder第一个元素为root，在inorder里面找到root，在它之前的为左子树(长stopIndex)，之后为右子树。preorder[1]到preorder[stopIndex]为左子树,之后为右子树，分别递归。

代码实现

//go
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func buildTree(preorder []int, inorder []int) *TreeNode {
    if len(preorder) == 0 {
        return nil
    }
    root := &TreeNode{preorder[0], nil, nil}
    i := 0
    // 在inorder里root的下标
    for ; i len(inorder); i++ {
        if inorder[i] == preorder[0] {
            break
        }
    }
    // root前的长度，即左子树的长度
    stopIndex := len(inorder[:i])+1
    root.Left = buildTree(preorder[1:stopIndex], inorder[:i])
    root.Right = buildTree(preorder[stopIndex:], inorder[i+1:])
    return root
}