解题思路
递归法
根据二叉树的前序遍历和中序遍历,根据前序遍历,可以找到第一个值就是根节点的值,到中序遍历中去寻找根节点,就可以确定左子树和右子树,由此递归进行,构建出完整二叉树。
上面说到需要在前序遍历中拿到根节点的值,然后去中序遍历中找这个值的位置。
所以为了方便期间,用哈希表建立中序遍历中值和下标的映射关系。
递归:
终止条件:
一棵树前序遍历的起点pre_left大于
一棵树前序遍历的终点pre_right就终止递归,返回空节点nullptr;
根据pre中的起点,即第一个值就是根节点的值,那么我们就可以构建出根节点了。
根据上面所说的映射关系,找到这个值在中序遍历的位置。
根据根节点在中序遍历的位置,我们就知道左子树中节点的个数了。
有了上面的准备工作,咱们就可以递归地去构建左右子树了,重要的就是左右子树前序遍历和中序遍历在pre和vin中的哪个范围上。
好了,最后返回root就可以了!
完整代码:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
unordered_map<int,int>m; //记录pre中的节点在vin中的位置,可以通过节点的值直接索引到在vin中的位置,建立映射关系
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
int n=pre.size();
//特殊情况,直接返回空树
if(n==0)return nullptr;
//为了方便起见,vin建立值和index的映射关系
for(auto node_val:pre){
for(int i=0;i<n;++i){
if(vin[i]==node_val)
m[node_val]=i;
}
}
return help(pre,vin,0,n-1,0,n-1);
}
//参数说明
//pre_left:一棵树前序遍历的起点 pre_right:一棵树前序遍历的终点
//vin_left:一棵树中序遍历的起点 vin_right:一棵树中序遍历的终点
TreeNode* help(vector<int> &pre,vector<int> &vin,int pre_left,int pre_right,int vin_left,int vin_right){
//终止条件
if(pre_left>pre_right)return nullptr;
//构造根节点
TreeNode* root=new TreeNode(pre[pre_left]);
//记录根节点在pre和vin中的索引
int pre_root_index=pre_left;
int vin_root_index=m[pre[pre_left]];
//记录根节点左子树中节点的个数,为了递归的时候定位下标
int left_node_count=vin_root_index-vin_left;
//递归构造左右节点
root->left=help(pre,vin,pre_left+1,pre_left+left_node_count,vin_left,vin_root_index-1);
root->right=help(pre,vin,pre_left+left_node_count+1,pre_right,vin_root_index+1,vin_right);
return root;
}
};