题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路
二叉树的前序遍历顺序是根节点、左子树、右子树
中序遍历顺序是左子树、根节点、右子树
前序遍历结果中首元素肯定是二叉树的根节点,要想构建出二叉树就要知道当前节点的左子树和右子树分别是谁,而中序遍历结果中根节点的左子树和右子树范围刚好是数组中当前节点的左区间和右区间
举例:前序遍历结果pre
{1,2,4,7,3,5,6,8},
中序遍历结果vin
{4,7,2,1,5,3,8,6}
根节点为前序遍历结果首元素1,则找到1在vin
中的下标为3,所以根节点1的左子树分别是4,7,2;右子树是5,3,8,6,然后我们遍历pre
,依次确定每一节点的左子树区间和右子树区间以及当前节点在其父节点的哪个区间内,如pre[1] = 2
,找到它在vin
数组中下标为2,在1的左子树区间内,所以确定2为1的左孩子节点,由此递归找下去,直到遍历完pre
代码演示
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
//找到前序遍历结果在中序遍历数组中对应的下标
int FindIndex(vector<int> vec,int start, int finish, int to_find)
{
for(;start < finish; start++)
{
if(vec[start] == to_find)
return start;
}
return -1;
}
TreeNode* _BulidTree(vector<int> pre,vector<int> vin, int left, int right,int* index)
{
if(left >= right)
return NULL;
//证明遍历完数组了
if(*index >= vin.size())
return NULL;
//从前序遍历数组中取出根节点
TreeNode* new_node = new TreeNode(pre[*index]);
//找到当前节点在中序遍历数组中的下标,以此来确定其左右子树区间
int cur = FindIndex(vin,left,right,pre[*index]);
(*index)++;
//左子树区间,左闭右开
new_node->left = _BulidTree(pre,vin,left,cur,index);
//右子树区间也同样
new_node->right = _BulidTree(pre,vin,cur+1,right,index);
return new_node;
}
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
int index = 0;//用来遍历前序数组
if(pre.size() != vin.size() || pre.size() == 0)
return NULL;
return _BulidTree(pre,vin,0,vin.size(),&index);
}
};