数据结构-根据前序和中序建立二叉树递归

一，题目

给定两个整数数组，preorder 和 postorder ，其中 preorder 是一个具有 无重复 值的二叉树的前序遍历，postorder 是同一棵树的后序遍历，重构并返回二叉树。

二，解题思路

1.前序遍历确定根节点：

前序遍历的第一个元素是整棵树的根节点。

2.中序遍历确定左右子树：

根据根节点在中序遍历中的位置，根节点左边的部分就是左子树，右边的部分就是右子树。

3.递归构建子树：

利用前序遍历找到根节点。
利用中序遍历确定左右子树的范围。
递归地构建左子树和右子树，直到所有节点都构建完毕。

三，详细步骤

Step 1: 初始化并建立辅助索引表

在开始之前，我们先遍历一遍中序遍历数组，将每个节点的值和其索引记录到一个哈希表（index_map）中。这使得我们可以在常数时间内确定某个值在中序遍历中的位置，从而快速分割左右子树。

Step 2: 从前序遍历中提取根节点

使用一个变量 pre 来追踪当前前序遍历数组中的位置（即第几个元素），每次递归时，pre 指向的元素即为当前子树的根节点。

Step 3: 递归构建子树

根据当前的 preorder[pre] 获取根节点，并根据该根节点的值在 index_map 中查找其在中序遍历中的索引 ind。

递归构建左子树和右子树：

左子树的范围为 in_left 到 ind - 1。
右子树的范围为 ind + 1 到 in_right。
每递归处理一次，pre 都自增1，以确保下次递归时处理下一个前序遍历的根节点。

Step 4: 递归终止条件

当 in_left > in_right 时，说明子树为空，返回 nullptr，终止递归。

Step 5: 输出验证

通过 printInOrder 函数来验证构建出的二叉树是否正确，通过中序遍历的方式输出结果与输入的中序遍历结果进行对比。

四， 源码

#include <iostream>
#include <vector>
#include <unordered_map>

using namespace std;

// 树节点结构体定义
struct TreeNode {
    int val;  // 节点值
    TreeNode* left;  // 左子树指针
    TreeNode* right;  // 右子树指针
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}  // 构造函数
};

unordered_map<int, int> index_map;  // 用于存储中序遍历中每个值的索引
int pre = 0;  // 用于标记当前处理的前序遍历位置

// 重建二叉树的递归函数
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder, int in_left, int in_right) {
    // 如果左边界大于右边界，说明子树为空
    if (in_left > in_right) {
        return nullptr;  // 返回空指针
    }

    // 获取当前前序遍历的根节点值
    int root_val = preorder[pre];
    pre++;  // 移动到下一个前序节点

    // 获取根节点在中序遍历中的位置
    int ind = index_map[root_val];

    // 创建当前根节点
    TreeNode* root = new TreeNode(root_val);

    // 递归构建左子树，范围是 in_left 到 ind-1
    root->left = buildTree(preorder, inorder, in_left, ind - 1);

    // 递归构建右子树，范围是 ind+1 到 in_right
    root->right = buildTree(preorder, inorder, ind + 1, in_right);

    // 返回当前构建的子树
    return root;
}

// 构建二叉树的辅助函数，初始化索引并调用递归函数
TreeNode* buildTreeWrapper(vector<int>& preorder, vector<int>& inorder) {
    pre = 0;  // 初始化前序遍历索引
    // 构建中序遍历值到索引的映射
    for (int i = 0; i < inorder.size(); i++) {
        index_map[inorder[i]] = i;
    }
    // 调用递归函数构建二叉树
    return buildTree(preorder, inorder, 0, inorder.size() - 1);
}

// 中序遍历输出二叉树
void printInOrder(TreeNode* root) {
    if (root == nullptr) {
        return;  // 如果当前节点为空，直接返回
    }
    printInOrder(root->left);  // 递归遍历左子树
    cout << root->val << " ";  // 输出当前节点值
    printInOrder(root->right);  // 递归遍历右子树
}

int main() {
    int n;
    cout << "请输入节点数量: ";  // 提示输入节点数量
    cin >> n;  // 读取节点数量

    vector<int> preorder(n), inorder(n);  // 定义存储前序遍历和中序遍历的数组

    cout << "请输入前序遍历: ";  // 提示输入前序遍历
    for (int i = 0; i < n; i++) {
        cin >> preorder[i];  // 依次读取前序遍历节点值
    }

    cout << "请输入中序遍历: ";  // 提示输入中序遍历
    for (int i = 0; i < n; i++) {
        cin >> inorder[i];  // 依次读取中序遍历节点值
    }

    // 构建二叉树
    TreeNode* root = buildTreeWrapper(preorder, inorder);

    // 输出中序遍历结果
    cout << "重构的二叉树的中序遍历为: ";
    printInOrder(root);  // 中序遍历输出
    cout << endl;

    return 0;  // 程序正常结束
}