西农数据结构作业_由二叉树的中序层序重建二叉树

最新推荐文章于 2024-09-10 20:37:54 发布

一根想学会敲代码的薯条

最新推荐文章于 2024-09-10 20:37:54 发布

阅读量381

点赞数 1

分类专栏：西农数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/m0_63577901/article/details/128026282

版权

西农数据结构专栏收录该内容

11 篇文章 5 订阅

订阅专栏

该程序实现了从给定的二叉树中序和层序遍历序列重建二叉树，并进一步进行先序和后序遍历。首先读取输入的二叉树节点数及中序和层序遍历序列，然后使用递归的BuildTree函数根据给定序列重建二叉树。重建后的二叉树分别按先序和后序进行遍历，将结果存储到pre和post向量中，最后输出这两个遍历序列。

摘要由CSDN通过智能技术生成

给定一个二叉树的中序和层序输出，重建二叉树并按先序和后序输出。

#include <iostream>
#include <vector>
using namespace std;
struct TreeNode {
	int data;
	TreeNode* leftchild;
	TreeNode* rightchild;
	TreeNode(int n) {
		data = n;
	}
};
vector<int> ceng;
vector<int> mid;
vector<int> pre;
vector<int> post;
TreeNode* BuildTree(int cl, int cr, int ml, int mr);
void preOrder(TreeNode* r);
void PostOrder(TreeNode* r);

int main() {
	int num;
	cin >> num;
	int temp;
	for (int i = 0; i < num; i++) {
		cin >> temp;
		ceng.push_back(temp);
	}
	for (int i = 0; i < num; i++) {
		cin >> temp;
		mid.push_back(temp);
	}
	TreeNode* root = BuildTree(0, num - 1, 0, num - 1);

	preOrder(root);
	PostOrder(root);

	for (int i = 0; i < num - 1; i++) {
		cout << pre[i]<<" ";
	}
	cout << pre[num - 1] << endl;

	for (int i = 0; i < num - 1; i++) {
		cout << post[i] << " ";
	}
	cout << post[num - 1] << endl;
	
}

TreeNode* BuildTree(int cl, int cr, int ml, int mr)
{
	if(ml>mr)
		return nullptr;

	int i, j;
	int cnt = 0;
	for (i = cl; i <= cr; i++) {
		for ( j = ml; j <= mr; j++) {
			if (ceng[i] == mid[j]) {
				cnt = 1;
				break;
			}
		}
		if (cnt == 1)
			break;
	}

	TreeNode* r = new TreeNode(ceng[i]);
	r->leftchild = BuildTree(cl + 1, cr, ml, j - 1);
	r->rightchild = BuildTree(cl + 1, cr, j + 1, mr);

	return r;
	

}

void preOrder(TreeNode* r)
{
	if (r == NULL)
		return;
	pre.push_back(r->data);
	preOrder(r->leftchild);
	preOrder(r->rightchild);

}

void PostOrder(TreeNode* r)
{
	if (r == NULL)
		return;

	PostOrder(r->leftchild);
	PostOrder(r->rightchild);
	post.push_back(r->data);

}