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

ShellDawn

于 2019-09-25 00:13:36 发布

阅读量89

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/101321992

leetcode 专栏收录该内容

251 篇文章

订阅专栏

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

摘要生成于 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* 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());
    }
};