105. 从前序与中序遍历序列构造二叉树/C++

最新推荐文章于 2022-02-03 21:05:33 发布

原创最新推荐文章于 2022-02-03 21:05:33 发布 · 174 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode/C++ 专栏收录该内容

248 篇文章

订阅专栏

本文介绍了一种使用前序遍历和中序遍历构建二叉树的方法。通过找到根节点和左右子树，然后递归地构建整棵树。具体实现包括定义二叉树节点结构，以及构建树的函数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
就是在两种序列中，找到根节点与左右子树，之后就是递归了。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        return build(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
    }
    TreeNode* build(vector<int>& preorder, int startPre, int endPre, vector<int>& inorder, int startIn, int endIn){
        if(startPre>endPre || startIn>endIn) return nullptr;
        
        int rootVal = preorder[startPre];
        TreeNode* root = new TreeNode(rootVal);
        if(startPre==endPre) return root;
        
        int rootIn = startIn;
        while(rootIn<=endIn && inorder[rootIn]!=rootVal)
            ++rootIn;
        
        int lenLeft = rootIn - startIn;
        root->left = build(preorder,startPre+1,startPre+lenLeft,inorder,startIn,rootIn-1);
        root->right = build(preorder,startPre+lenLeft+1,endPre,inorder,rootIn+1,endIn);
        return root;
    }
};