剑指 Offer 07. 重建二叉树

1.题目:

输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

 

示例 1:

 

 

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]
 

限制:

0 <= 节点个数 <= 5000

 

2. 分析与代码:

重构二叉树看另一篇博客:重构二叉树 + 输出 (层序遍历、其他遍历) - 湘summer - 博客园 (cnblogs.com)

代码:

复制代码
/**
 * 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 getBinaryTree(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);
    }

    TreeNode* getBinaryTree(vector<int>& preorder,int preBegin,int preEnd,vector<int>& inorder,int inBegin,int inEnd){
        if(preBegin > preEnd)
            return NULL;
        TreeNode* tree = new TreeNode(preorder[preBegin]);
        // *tree = (preorder[preBegin],NULL,NULL);
        int r = findRootInInorder(inorder,inBegin,inEnd,preorder[preBegin]);
        int count = r-inBegin; //inBegin从0开始,无需加1
        if(r!=inBegin)
            tree->left = getBinaryTree(preorder,preBegin+1,preBegin+count,inorder,inBegin,r-1);
        if(r!=inEnd)
            tree->right = getBinaryTree(preorder,preBegin+count+1,preEnd,inorder,r+1,inEnd);
        return tree;
    }

    int findRootInInorder(vector<int>& inorder,int inBegin,int inEnd,int key){
        for(int i=inBegin;i<=inEnd;i++){
            if(inorder[i] == key)
                return i;
        }
        return 0;
    }
};
复制代码

 

也可以用hashMap去查找根节点在中序遍历中的位置。

posted @   湘summer  阅读(15)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· DeepSeek 开源周回顾「GitHub 热点速览」
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
点击右上角即可分享
微信分享提示