根据前序遍历和中序遍历恢复二叉树
在二叉树中,我们可以通过前序遍历和中序遍历恢复出一个完整的二叉树。是因为在恢复的过程中,我们通过前序遍历找到树的根节点,通过根节点和中序遍历找到二叉树的左右子树,然后再进行递归就可以。
例如 某一个树的前序遍历:{1,2,4,7,3,5,6,8}
某一个树的中序遍历:{4,7,2,1,5,3,8,6}
通过前序遍历找到输的根节点为1,在中序遍历中,以1位分割的就是左右子树。此时,我们再看前序遍历中除1以外的其他数据,2,4,7就是他的左子树,并且左子树的根节点为2,3,5,6,8就是右子树,3为根节点。重复上述操作,我们就可以完成二叉树的恢复。
题目来源:牛客网:重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
class Solution {
public:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
int len1 = pre.size();
int len2 = vin.size();
if(len1 == 0){
return NULL;
}
vector<int> pre_left,pre_right,vin_left,vin_right;
TreeNode* head = new TreeNode(pre[0]);
int index = 0;
for(int i = 0; i < len2;i++){
if(vin[i] == pre[0]){
index = i;
break;
}
}
for(int i = 0; i < index;i++){
vin_left.push_back(vin[i]);
pre_left.push_back(pre[i+1]);
}
for(int i = index+1; i < len2;i++){
vin_right.push_back(vin[i]);
pre_right.push_back(pre[i]);
}
head->left = reConstructBinaryTree(pre_left, vin_left);
head->right = reConstructBinaryTree(pre_right, vin_right);
return head;
}
};
在这里采用递归的方式进行还原,最后返回二叉树的根节点