No.120 - LeetCode105 - (经典) - 前序中序生成树

最新推荐文章于 2025-07-26 00:32:18 发布

原创最新推荐文章于 2025-07-26 00:32:18 发布 · 104 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用前序遍历和中序遍历构建二叉树的方法。通过递归地划分前序和中序数组来确定每个节点的位置，并创建相应的二叉树节点。

/**
 * 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* dfs(vector<int>& pre,int preL,int preR,vector<int>& in,int inL,int inR){
        if(preL >= preR) return NULL;
        int loc = inL;
        for(;loc<inR;loc++) if(pre[preL] == in[loc]) break;
        TreeNode* now = new TreeNode(pre[preL]);
        int lenth = loc - inL;
        now->left = dfs(pre,preL+1,preL+lenth+1,in,inL,loc);
        now->right = dfs(pre,preL+lenth+1,preR,in,loc+1,inR);
        return now;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        return dfs(preorder,0,preorder.size(),inorder,0,inorder.size());
    }
};