剑指 Offer 07. 重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例 1:

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]

限制：

0 <= 节点个数 <= 5000

---

        本题是要用前序加中序来确定一个二叉树，前序的规则是根左右，所以在前序数组中第一个数肯定是根节点，但是后面哪些是左子树和右子树需要通过中序来确定。中序的规则是左根右，所以根据前序找到根节点，然后在中序数组中找到根节点对应的位置，该位置左侧为左子树、右侧为右子树。二叉树的构建规则大概就是这样，然后就是通过分治的思想来递归地将二叉树建立起来。

        递归过程大致如下：

1. 首先根据前序数组来找到根节点
2. 找到根节点在中序数组中的位置
3. 中序数组中根节点左边的都是左子树，右边的都是右子树，然后根据序列长度在前序中找出左子树和右子树的范围。
4. 开启左子树和右子树的递归，最后连接到根节点上返回。

        过程中有一步是找到根节点在中序数组中的位置，这个可以通过map来实现。代码如下：

class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        def recur(pos,l_pos,r_pos):
            if l_pos > r_pos:
                return
            node=TreeNode(preorder[pos])
            i=m[preorder[pos]]
            node.left=recur(pos+1,l_pos,i-1)
            node.right=recur(pos+1+i-l_pos,i+1,r_pos)
            return node
        m,inorder={},inorder
        for i in range(len(inorder)):
            m[inorder[i]]=i
        return recur(0,0,len(preorder)-1)

        递归函数的三个参数的意义分别是：当前根节点在先序数组中的位置、当前根节点下所有元素在中序数组中范围的左边界、根节点下所有元素在中序数组中范围的右边界。

        进入循环后先判断左右边界是否交叉了，如果交叉说明越过了叶子结点可以直接返回了。

        然后建立当前的根节点，找到根节点在中序数组中的位置，也就是 i 。

        然后开启左右子树的递归，pos+1是左子树的根节点在前序数组中的位置，pos+1+i-l_pos是右子树的根节点在前序数组中的位置（可以理解为根节点+左子树的序列长度+1）

        然后将左右子树连接到根节点上返回。