给出某二叉树的前序遍历和中序遍历的序列,请重建出该二叉树:
根据二叉树的前序遍历的特性,每次读取的第一个值一定是根节点,这样我们可以在中序遍历的序列中找到当前的根节点的位置。该位置左边的就是该树的左子树,右边的就是右子树了。
因此,我们每次都需要在前序遍历中找根节点并创建一个根节点,然后在中序遍历中确定根节点位置,并确定当前根节点的左右子树,然后以同样的方法去构建左右子树。这整个过程其实是一个递归的过程。
代码实现如下:
TreeNode* rebuildTree(int pre[], int mid[], int len)
{
if(0 == len)
return NULL;
int i = 0;
//在中序序列中找到前序第一个节点
for(; i< len; i++){
if(mid[i] == pre[0])
break;
}
//新建root节点
TreeNode* root = new TreeNode(pre[0]);
int pre_left[i] = {0};
int mid_left[i]={0};
int pre_right[len-i-1] = {0};
int mid_right[len-i-1]={0};
//左子树的前序和中序序列
for(int j =0;j<i;j++){
pre_left[j] = pre[j+1];
mid_left[j] = mid[j];
}
//右子树的前序和中序序列
for(int q=0,int k =i+1;k<len;k++,q++){
pre_right[q] = pre[k]; mid_right[q] = mid[k]; }
//递归完成
root->left = rebuildTree(pre_left, mid_left, i);
root->right = rebuildTree(pre_right, mid_right, len-i-1);
return root;
}