C++实现二叉树先序遍历、中序遍历、后序遍历、利用先序遍历序列和中序遍历序列构造二叉树

#include <cstdio>
#include <string>
#include <iostream>
#include <stack>
using namespace std;
typedef struct BiTNode {
	char data;
	BiTNode* lchild = nullptr, * rchild = nullptr;
}*BiTree; //【BiTNode 二叉树结点】【BiTree 二叉树】
BiTree CreatBiTree(string preStr, string inStr) { //根据先序遍历序列和中序遍历序列构造二叉树
	BiTree temp = new BiTNode;
	temp->data = *(preStr.cbegin());
	size_t i;
	for (i = 0; i < inStr.size(); i++)
		if (inStr[i] == temp->data) break;
	string LpreStr, LinStr, RpreStr, RinStr;
	LpreStr = preStr.substr(1, i);
	LinStr = inStr.substr(0, i);
	RpreStr = preStr.substr(i + 1);
	RinStr = inStr.substr(i + 1);
	if (!LpreStr.empty())
		temp->lchild = CreatBiTree(LpreStr, LinStr);
	if (!RpreStr.empty())
		temp->rchild = CreatBiTree(RpreStr, RinStr);
	return temp;
}
string PreOrder(BiTree B) { //获取先序遍历序列
	if (B != nullptr) return B->data + PreOrder(B->lchild) + PreOrder(B->rchild);
	else return "";
}
string InOrder(BiTree B) { //获取中序遍历序列
	if (B != nullptr) return PreOrder(B->lchild) + B->data + PreOrder(B->rchild);
	else return "";
}
string InOrder2(BiTree B) {  //利用 栈 获取中序遍历序列
	string result;
	BiTNode* temp = B->lchild;
	stack<BiTNode*> st;	st.push(B);
	while (temp != nullptr) {
		st.push(temp);
		temp = temp->lchild;
	}
	while (!st.empty()) {
		temp = st.top(); st.pop();
		result += temp->data;
		temp = temp->rchild;
		if (temp != nullptr) {
			st.push(temp);
			temp = temp->lchild;
			while (temp != nullptr) {
				st.push(temp);
				temp = temp->lchild;
			}
		}
	}
	return result;
}
string PostOrder(BiTree B) {  //获取后序遍历序列
	if (B != nullptr) return PreOrder(B->lchild) + PreOrder(B->rchild) + B->data;
	else return "";
}
int main()
{
	string pre("ABDGCEF"), in("DGBAECF");
	auto MyBiTree = CreatBiTree(pre, in);
	cout << InOrder2(MyBiTree) << endl;
	return 0;
}
  • 3
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
我们可以利用递归的思想来解决这个问题。一个二叉树先序遍历序列的第一个元素一定是根节点,而在中序遍历序列中,根节点的左侧都是左子树的节点,右侧都是右子树的节点。因此,我们可以根据这个性质来构建二叉树,并且递归地处理左右子树。 具体的实现步骤如下: 1. 根据先序遍历序列的第一个元素创建根节点。 2. 在中序遍历序列中找到根节点的位置,将序列分成左右两部分。 3. 递归地处理左子树,将左子树的先序遍历序列中序遍历序列作为参数传入。 4. 递归地处理右子树,将右子树的先序遍历序列中序遍历序列作为参数传入。 5. 将根节点的值加入到后序遍历序列中。 6. 返回后序遍历序列。 下面是 C++ 的代码实现: ```cpp #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) {} }; class Solution { public: vector<int> buildTree(vector<int>& preorder, vector<int>& inorder) { // 将中序遍历序列的每个元素及其下标存储到哈希表中 for (int i = 0; i < inorder.size(); i++) { index[inorder[i]] = i; } return buildTree(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1); } private: unordered_map<int, int> index; // 根据先序遍历序列中序遍历序列构建二叉树,并返回后序遍历序列 vector<int> buildTree(vector<int>& preorder, int preStart, int preEnd, vector<int>& inorder, int inStart, int inEnd) { if (preStart > preEnd) { return {}; } // 先序遍历序列的第一个元素是根节点 int rootVal = preorder[preStart]; TreeNode* root = new TreeNode(rootVal); // 在中序遍历序列中找到根节点的位置 int rootIndex = index[rootVal]; // 计算左子树的长度 int leftSize = rootIndex - inStart; // 递归处理左子树 vector<int> left = buildTree(preorder, preStart + 1, preStart + leftSize, inorder, inStart, rootIndex - 1); // 递归处理右子树 vector<int> right = buildTree(preorder, preStart + leftSize + 1, preEnd, inorder, rootIndex + 1, inEnd); // 将根节点的值加入到后序遍历序列中 vector<int> res = left; res.insert(res.end(), right.begin(), right.end()); res.push_back(rootVal); return res; } }; ``` 时间复杂度为 O(n),其中 n 是二叉树中的节点个数。这是因为我们需要遍历每个节点恰好一次。空间复杂度为 O(n),其中 n 是二叉树中的节点个数。除了存储答案所需的空间,我们还需要 O(n) 的空间存储哈希表。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值