由前序遍历和中序遍历重建二叉树

思路:

  1. 根据前序遍历序列找到根节点的值
  2. 在中序遍历序列中找到根节点的位置
  3. 找到左子树根节点(中序遍历中根结点的前一个位置,如果存在的话,就是左子树的根结点)
  4. 找到右子树根节点(中序遍历中根结点的后一个位置,如果存在的话,就是右子树的根结点)
  5. 递归地重建子树(使用左子树的前序、中序序列构建左子树;使用右子树的前序、中序序列构建右子树)

例如下面这张图(出自《剑指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;
}
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值