标签：preorder 遍历 Offer int root 007 二叉树 inorder 节点

题目描述

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

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

• 示例
  

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

递归

划分为更小规模子问题思想：

1. 找出根节点：前序遍历序列第一个节点为当前树的根节点
2. 中序遍历序列中划分子树：中序遍历序列中以根节点为分界，左侧的序列节点全为左子树节点，右侧的序列节点全为右子树的节点
3. 前序遍历序列中划分子树：利用中序遍历序列中计算出左子树节点数目和右子树节点数目，应用到前序遍历序列中，将前序遍历序列也划分为左子树、右子树、当前树根节点
4. 划分更小规模的子问题：左子树的前序、中序，右子树的前序、中序

递归退出条件：当前节点为叶子节点

// C++
struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder)
{
    if (preorder.empty() || inorder.empty())
        return nullptr;

    return buildTreeCore(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1);
}

TreeNode *buildTreeCore(vector<int> &preorder, int startPreorder, int endPreorder, vector<int> &inorder, int startInorder, int endInorder)
{
    // 构建根节点
    int rootValue = preorder[startPreorder];
    TreeNode *rootNode = new TreeNode(rootValue);
    rootNode->val = rootValue;
    rootNode->left = nullptr;
    rootNode->right = nullptr;

    // 子树只有一个节点
    if (startPreorder == endPreorder && startInorder == endInorder)
        return rootNode;

    // 寻找根节点在中序遍历中的索引位置
    int rootInorder = startInorder;
    while (inorder[rootInorder] != rootValue)    // 可以用 index 系统函数找出索引
        ++rootInorder;

    int leftLength = rootInorder - startInorder;    // 左子树节点个数
    // 构建左子树
    if (leftLength > 0)
        rootNode->left = buildTreeCore(preorder, startPreorder + 1, startPreorder + leftLength, inorder, startInorder, rootInorder - 1);

    
    int rightLength = endInorder - rootInorder;     // 右子树节点个数
    // 构建右子树
    if (rightLength > 0)
        rootNode->right = buildTreeCore(preorder, startPreorder + leftLength + 1, endPreorder, inorder, rootInorder + 1, endInorder);

    return rootNode;
}

# python
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


def buildTree(preorder: List[int], inorder: List[int]) -> TreeNode:
    if len(preorder) == 0 or len(inorder)==0 :
        return None
    return buildTreeCore(preorder, 0, len(preorder) - 1, inorder, 0, len(inorder)-1)


def buildTreeCore(preorder: List[int], start_preorder: int, end_preorder: int, inorder: List[int], start_inorder, end_inorder):
    # 创建根节点
    root_value = preorder[start_preorder]
    root_node = TreeNode(root_value)
    # 构造函数已经定义为了 None，不需要重复
    # root_node.left = None
    # root_node.right = None

    # 退出递归条件：子树只有一个节点
    if start_preorder == end_preorder and start_inorder == end_inorder:
        return root_node

    # 在中序遍历序列中寻找根节点位置
    # root_inorder_index = start_inorder
    # while inorder[root_inorder_index] != root_value:
    #     ++root_inorder_index
    root_inorder_index = inorder.index(root_value)

    # 构建左子树
    left_length = root_inorder_index - start_inorder
    if left_length > 0:
        root_node.left = buildTreeCore(preorder, start_preorder+1, start_preorder+left_length,inorder,start_inorder,root_inorder_index-1)

    # 构建右子树
    right_length = end_inorder - root_inorder_index
    if right_length > 0 :
        root_node.right = buildTreeCore(preorder,start_preorder + left_length + 1,end_preorder, inorder, root_inorder_index+1,end_inorder)

    return root_node

标签：preorder,遍历,Offer,int,root,007,二叉树,inorder,节点
来源： https://www.cnblogs.com/cscshi/p/15812192.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。