剑指offer—重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
博客中代码均在牛客C++11(clang++ 3.9)中通过
第一次做这个题真的毫无头绪,主要还是没有理解前序遍历和中序遍历的特点,后来看到牛客上写的代码方才豁然开朗,接下来就具体讲解一下
首先要知道,前序遍历第一个数字就是就是二叉树根节点的值,而中序遍历中根节点位于中间,其左边是左子树的值,右边是右子树的值,结合这一特性来实现重建二叉树。
- 我们可以自己写一个函数,在传入两个序列的同时传入两个序列的起止位置下标(主要用来用来控制我们构建左右子树的范围),当某一个序列的起始位置大于终止位置时返回nullptr
- 在函数内部,首先用前序遍历第一个数字来给创建的根节点赋值,并在中序遍历序列中寻找与该数字相同的位置即为中序遍历中根节点的位置
- 找到之后,在其左右分别是左右子树的中序序列,此时我们就可以将问题分解为重建两个小二叉树,利用递归的方法不断深入来逐步构建二叉树的每一个节点,实现二叉树的重建
代码如下:
class Solution {
public:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
// 如果前序遍历货值中序遍历数组为空,直接返回nullptr
if(0 == pre.size() || 0 == vin.size())
return nullptr;
// 重新写函数,传入两种遍历的起始和终止位置的下标
TreeNode* ret = reConstructBinaryTree1(pre,0,pre.size()-1,vin,0,vin.size()-1);
return ret;
}
public:
TreeNode* reConstructBinaryTree1(vector<int> pre,int p1,int p2,vector<int> vin,int v1,int v2)
{
// 当前序后中序遍历起始位置大于终止位置时,返回nullptr
if(p1>p2 || v1>v2)
return nullptr;
// 定义根节点,开辟空间并用前序遍历第一个值来赋值
TreeNode* root = new TreeNode(pre[p1]);
// 在中序遍历中循环查找根节点位置,递归构建左右子树的根节点
for(int i=v1;i<=v2;++i)
{
if(pre[p1] == vin[i])
{
// 找到根节点后构建左子树,传入前序遍历中左子树(根节点后i-v1个元素)的起止位置和中序遍历中左子树(根节点左边数组)的起止位置
root->left = reConstructBinaryTree1(pre,p1+1,p1+i-v1,vin,v1,i-1);
// 找到根节点后构建左子树,传入前序遍历中右子树(根节点后i-v1+1开始到终止位置)的起止位置和中序遍历中右子树(根节点右边数组)的起止位置
root->right = reConstructBinaryTree1(pre,p1+i-v1+1,p2,vin,i+1,v2);
}
}
return root;
}
};