【数据结构与算法】根据后序遍历序列和中序遍历序列构造二叉树 C++实现（二叉树+二叉链表+递归）

根据后序遍历序列和中序遍历序列构造二叉树

题目描述

给定一棵二叉树的后序遍历序列 post[s1..t1] 和中序遍历序列 in[s2..t2]，设二叉树采用二叉链表存储，设计一个算法构造这棵二叉树。请使用递归方法解决。

输入格式

输入包括两行：

• 第一行为后序遍历序列 post[s1..t1]。
• 第二行为中序遍历序列 in[s2..t2]。

保证构成一棵唯一的二叉树。

输出格式

输出为前序遍历序列。

样例 #1

样例输入 #1

DEBGFCA
DBEAFGC

样例输出 #1

ABDECFG

提示

首先，后序遍历的最后一个元素是树的根。然后，你可以在中序遍历序列中找到这个根，这样就可以确定哪些元素在左子树，哪些在右子树。然后，你可以递归地构造左子树和右子树。

---

思路

函数 createBiTree 是构造二叉树的主要算法。参数包括后序遍历序列 last，中序遍历序列 mid，和序列的长度 len。

一开始判断长度 len 是否小于等于0，如果是，返回 NULL，表示这是一个空的二叉树。

接着，从后序遍历序列 last 中取出最后一个元素，即根节点 root。然后在中序遍历序列 mid 中找到根节点的位置 index。这个位置将中序遍历序列分为两部分，前半部分是左子树的中序遍历序列，后半部分是右子树的中序遍历序列。

然后动态分配内存，创建一个新的二叉树节点 T，并将 root 存入 T->data。

接下来，递归地调用 createBiTree，构造左子树和右子树。左子树的后序遍历序列和中序遍历序列分别是 last 和 mid 的前 index 个元素，右子树的后序遍历序列和中序遍历序列分别是 last 和 mid 的后 len - index - 1 个元素。将构造出的左子树和右子树分别赋给 T->left 和 T->right。

最后返回构造出的二叉树 T。

算法分析

这个函数的时间复杂度为 $O(n^2)$，因为在每一层递归中，都需要在中序遍历序列中找到根节点的位置，这需要遍历整个中序遍历序列，所以时间复杂度为 $O(n)$，而递归的深度最多为 $n$，所以总的时间复杂度为 $O(n^2)$。空间复杂度为 $O(n)$，主要是存储构造出的二叉树和递归调用的栈空间。

---

AC代码

#include <algorithm>
#include <cstring>
#include <iostream>
#define AUTHOR "HEX9CF"
using namespace std;
using Status = int;
using ElemType = char;

const int N = 1e6 + 7;
const int TRUE = 1;
const int FALSE = 0;
const int OK = 1;
const int ERROR = 0;
const int INFEASIBLE = -1;
// const int OVERFLOW = -2;

int n;
ElemType a[N];

struct TreeNode {
	ElemType data;
	struct TreeNode *left, *right;
};
using BiTree = TreeNode *;

BiTree createBiTree(char *last, char *mid, int len) {
	if (len <= 0) {
		return NULL;
	}
	// 后序序列最后一个节点为根节点
	char root = last[len - 1];
	// 中序序列中找根节点
	int index = 0;
	while (mid[index] != root) {
		index++;
	}

	// 创建根节点
	BiTree T = (TreeNode *)malloc(sizeof(TreeNode));

	T->data = root;
	T->left = createBiTree(last, mid, index);
	T->right = createBiTree(last + index, mid + index + 1, len - index - 1);
	return T;
}

void preorder(BiTree T) {
	if (T) {
		cout << T->data;
		preorder(T->left);
		preorder(T->right);
	}
}

void inorder(BiTree T) {
	if (T) {
		inorder(T->left);
		cout << T->data;
		inorder(T->right);
	}
}

void postorder(BiTree T) {
	if (T) {
		postorder(T->left);
		postorder(T->right);
		cout << T->data;
	}
}

int main() {
	char post[N], in[N];
	cin >> post >> in;

	BiTree T = createBiTree(post, in, strlen(post));

	preorder(T);
	cout << "\n";
	// inorder(T);
	// cout << "\n";
	// postorder(T);
	// cout << "\n";
	return 0;
}