从前序与中序遍历中构造二叉树---leetcode105题

最新推荐文章于 2025-08-04 19:56:00 发布

原创最新推荐文章于 2025-08-04 19:56:00 发布 · 145 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #深度优先 #算法

数据结构与算法学习记录专栏收录该内容

79 篇文章

订阅专栏

该博客介绍了如何利用先序遍历和中序遍历序列重建一棵二叉树，通过递归实现从给定的整数数组中构造二叉树的根节点。关键步骤包括找到中序遍历中的根节点位置，然后递归地构造左右子树。

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

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* traversal(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() == 0) return NULL;

        int rootval = preorder[0];
        TreeNode* root = new TreeNode(rootval);

        if(preorder.size() == 1) return root;
        int deliminterIndex;
        for(deliminterIndex = 0; deliminterIndex < inorder.size(); deliminterIndex++){
            if(inorder[deliminterIndex] == rootval) break;
        }

        vector<int> leftinorder(inorder.begin(), inorder.begin() + deliminterIndex);
        vector<int> rightinorder(inorder.begin() + deliminterIndex + 1, inorder.end());

        vector<int> leftpreorder(preorder.begin() + 1, preorder.begin()+leftinorder.size()+1);
        vector<int> rightpreorder(preorder.begin()+leftinorder.size()+1,preorder.end());

        root->left = traversal(leftpreorder,leftinorder);
        root->right = traversal(rightpreorder,rightinorder);

        return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() == 0 || inorder.size() == 0) return NULL;
        return traversal(preorder,inorder);    
    }
};