剑指 Offer 07. 重建二叉树

不正经小新

已于 2023-09-08 07:59:00 修改

阅读量3.4w

点赞数 2

分类专栏：数据结构与算法文章标签：算法 leetcode 深度优先后端

于 2023-02-05 20:01:44 首次发布

本文链接：https://blog.csdn.net/m0_46833224/article/details/128893029

版权

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

63 篇文章 1 订阅

订阅专栏

⭐简单说两句⭐

CSDN个人主页：后端小知识

🔎GZH：后端小知识

🎉欢迎关注🔎点赞👍收藏⭐️留言📝

Hello吖，各位小伙伴好久不见，十分想念，已经有一段时间没有更新博客了，具体原因嘛（放在文末）

题目：剑指 Offer 07. 重建二叉树，我们今天来看一道递归的题目吧，这是选自剑指Offer上的一道题，好了，我们一起来看看题意吧：

考虑到直接复制题目，或者截屏的方式不是很方便阅读，我就把直接题目链接放下面！
题目传送门：剑指 Offer 07. 重建二叉树

思路:

可以先看看代码（有注释的），然后再来看看这个思路

本题的一个难点就是：找到左右子树在前序和中序遍历序列中对应的区间

在前序序列中取得根节点（序列第一个），然后在中序序列中找到根节点的位置记为pos,

中序序列pos左边的则为左子树，对应数量为 pos - 中序序列起始位置（inStart) ,

将中序序列pos左边的节点数量记录为x，x = pos - inStart

二叉树的左子树在前序序列中的区间为：preStart（前序序列中的第一个位置）+1到preStart+x，在中序序列中的位置为inStart（中序序列中的第一个位置）到pos-1

二叉树的右子树在前序序列中的区间为：preStart+x+1,到preEnd（前序序列的最后一个位置），在中序序列中的位置为pos+1到inEnd（中序序列中的最后一个位置）

递归处理以上步骤

我们来看看成功AC的代码吧：

class Solution {
    vector<int> pre,in;
public:
    TreeNode* dfs(int preStart,int preEnd,int inStart,int inEnd){
        if(preStart>preEnd||inStart>inEnd) return NULL;
        //前序遍历中的开始节点就是根节点，取出根节点值
        int t = pre[preStart];
        //寻找子树的根节点在中序序列中的哪个位置
        int pos;
        for(int i=0;i<in.size()-1;i++){ if(in[i]==t){ pos = i; break; }}
        auto s = new TreeNode(t);
        //【pos-inStart】为左子树中的节点数
        int x = pos - inStart;
        //递归构建左子树，并连上根节点
        //前序遍历中的区间【preStart+1,preStart+x】，中序遍历中的区间【inStart,pos-1】
        s->left = dfs(preStart+1,preStart+x,inStart,pos-1);
        //递归构建右子树，并连上根节点
        //前序遍历中的区间【preStart+1+x,preEnd】,中序遍历中的区间【pos+1，inEnd】
        s->right = dfs(preStart+1+x,preEnd,pos+1,inEnd);
        return s;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        pre = preorder; in = inorder;
        return dfs(0,pre.size()-1,0,in.size()-1);
    }
};