C++详解:剑指Offer 07. 重建二叉树

    技术2022-07-12  86

    原题LeetCode链接:剑指Offer 07. 重建二叉树

    题目:剑指Offer 07. 重建二叉树

    输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如,给出

    前序遍历 preorder = [3,9,20,15,7] 中序遍历 inorder = [9,3,15,20,7]

    返回如下二叉树

    3 / \ 9 20 / \ 15 7

    思路

    ​ 给出两个遍历序列来建二叉树这种题目解决方法几乎是一致的。

    ​ 1.前序遍历序列的第一个元素是整棵树的根节点

    ​ 2.在中序遍历中找到根节点,它的左边就是左子树的元素,右边就是右子树的元素

    ​ 3.计算出左子树和右子树各有多少节点,然后继续在先序遍历中寻找左右子树的根节点,递归执行下去,直到完成。

    C++代码

    class Solution { public: TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) { return build(preorder, inorder, 0, preorder.size()-1, 0, inorder.size()-1); } //类似建树的题目都需要另外写一个函数 //preL,PreR,inL,inR分别是前序遍历起始和结束位置、后序遍历起始和结束位置 TreeNode* build(vector<int>& preorder, vector<int>& inorder, int preL, int preR, int inL, int inR){ if(preL > preR || inL > inR) return NULL; TreeNode* root = new TreeNode(preorder[preL]); int i; for(i = inL; i < inR; i++){ if(preorder[preL] == inorder[i]) break; } root->left = build(preorder, inorder, preL+1, preL + i - inL, inL, i-1); root->right = build(preorder, inorder, preL+i-inL+1, preR, i+1, inR); return root; } };

    复杂度分析

    时间复杂度:O(n).每个结点都要经历一遍建树过程

    空间复杂度:O(n).存储整棵树

    Processed: 0.019, SQL: 10