LeetCode 剑指 Offer 07. 重建二叉树

最新推荐文章于 2024-09-27 15:00:48 发布

暮雨林钟

最新推荐文章于 2024-09-27 15:00:48 发布

阅读量188

点赞数

分类专栏： LeetCode刷题记录文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/m0_66776925/article/details/123190887

版权

LeetCode刷题记录专栏收录该内容

461 篇文章 1 订阅

订阅专栏

具体思路：

递归分治；

具体代码：

/**
 * 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:
    void fun(int lpre,int rpre,int lin,int rin,vector<int>& pre,vector<int>& in,TreeNode* &root){
        if(lin>rin)
            return;
        if(lin==rin){
            root=new TreeNode(in[lin]);
            return;
        }
        int mid=-1;
        for(int i=lin;i<=rin;i++){
            if(in[i]==pre[lpre]){
                mid=i;
                break;
            }
        }
        root=new TreeNode(in[mid]);
        int left=mid-lin;
        int right=rin-mid;
        if(mid>lin){
            fun(lpre+1,lpre+left,lin,mid-1,pre,in,root->left);
        }
        if(mid<rin){
            fun(lpre+left+1,lpre+left+right,mid+1,rin,pre,in,root->right);
        }
    }

    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        TreeNode* root=NULL;
        fun(0,inorder.size()-1,0,preorder.size()-1,preorder,inorder,root);
        return root;
    }
};