思路:
- 根据前序遍历序列找到根节点的值
- 在中序遍历序列中找到根节点的位置
- 找到左子树根节点(中序遍历中根结点的前一个位置,如果存在的话,就是左子树的根结点)
- 找到右子树根节点(中序遍历中根结点的后一个位置,如果存在的话,就是右子树的根结点)
- 递归地重建子树(使用左子树的前序、中序序列构建左子树;使用右子树的前序、中序序列构建右子树)
例如下面这张图(出自《剑指offer》):
下面是代码:
/* 由前序遍历序列和中序遍历序列重建二叉树 */
#include <iostream>
#include <cstdlib>
using namespace std;
struct Node
{
int val;
Node* left;
Node* right;
};
Node* construct(int* preOrder, int* inOrder, int length)
{
Node* root = (Node*)malloc(sizeof(Node)); // 为子树的根节点分配空间(我们把整棵树的根节点也暂且当作子树)
int rootVal = preOrder[0]; // 子树的根结的值一定保存在前序遍历序列的第一个位置
root->val = rootVal; // 设置子树根结点的值
root->left = root->right = nullptr; // 先将左右子树均设置为空指针,避免出现野指针
// 如果前序遍历和中序遍历序列的长度为1,说明只有一个结点,直接作为叶结点返回即可
if (1 == length) return root;
// rootIn用于保存当前子树的根节点在中序遍历序列中的位置
// rootIn-1(如果存在的话)就是当前子树的左子树的根节点
// rootIn+1(如果存在的话)就是当前子树的右子树的根节点
int* rootIn = inOrder;
while (rootIn <= inOrder && *rootIn != rootVal) // 查找当前子树的根节点在中序遍历序列中的位置
++rootIn;
int lenL = rootIn - inOrder; // 左子树的长度
int lenR = length - lenL - 1; // 右子树的长度
if (lenL > 0) // 如果当前子树的左子树存在,递归的构建左子树
root->left = construct(preOrder + 1, inOrder, lenL);
if (lenR > 0) // 如果当前子树的右子树存在,递归的构建左子树
root->right = construct(preOrder + lenL + 1, rootIn + 1, lenR);
return root; // 返回当前子树的根节点
}
void tranverse(Node* root) // 前序遍历重建得到的二叉树
{
if (nullptr == root) return;
cout << root->val << ' ';
tranverse(root->left);
tranverse(root->right);
}
int main()
{
int preOrderSeq[] = {1, 2, 4, 7, 3, 5, 6, 8};
int inOrderSeq[] = {4, 7, 2, 1, 5, 3, 8, 6};
const int length = 8;
Node* root = construct(preOrderSeq, inOrderSeq, length);
tranverse(root);
return 0;
}