OJ_重建二叉树

Listennnnnnnnn

已于 2024-03-09 12:25:07 修改

阅读量711

点赞数 25

分类专栏：数据结构与算法文章标签： c语言 c++ 算法

于 2024-03-01 14:04:03 首次发布

本文链接：https://blog.csdn.net/Listennnn/article/details/136393426

版权

数据结构与算法专栏收录该内容

91 篇文章 2 订阅

订阅专栏

本文介绍了如何通过C++编程实现根据给定的二叉树先序遍历和中序遍历序列重构后序遍历序列的过程，包括Rebuild函数的递归逻辑和postOrder函数的后序遍历算法。

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

题干

已知：二叉树的先序序列和中序序列
求：后序序列

C++实现

思路

在先序里找根
拿根在中序里找根的位置
此时可以把中序分为左子树-根-右子树，然后就可以把先序分为根-左子树-右子树
然后再递归处理左子树部分和右子树部分

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<string>
using namespace std;

struct TreeNode
{
	char data;
	TreeNode* left;
	TreeNode* right;
};

TreeNode* Rebuild(string preOrder, string inOrder) {
	if (preOrder.size() == 0) {
		return NULL;
	}
	else {
		char rootdata = preOrder[0];//在先序中找到根
		TreeNode* pNew = new TreeNode;
		pNew->data = rootdata;
		int pos = inOrder.find(rootdata);//找到根在中序中的位置
		//preOrder.substr(1,pos)这是左子树的先序
		//inOrder.substr(0,pos)这是左子树的中序
		//preOrder.substr(pos+1,末尾)这是右子树的先序
		//inOrder.substr(pos+1,末尾)这是右子树的中序
		//函数原型：string substr(int pos = 0, int len = npos); //返回由pos开始的npos个字符组成的字符串
		//如果你只提供一个参数给substr()。在这种情况下，substr 函数将提取从该位置开始到字符串末尾的所有字符
		pNew->left = Rebuild(preOrder.substr(1, pos), inOrder.substr(0, pos));
		pNew->right = Rebuild(preOrder.substr(pos + 1), inOrder.substr(pos + 1));
		return pNew;
	}
}

void postOrder(TreeNode* proot) {
	if (proot == NULL) {
		return;
	}
	else {
		postOrder(proot->left);
		postOrder(proot->right);
		printf("%c", proot->data);
	}
}

int main() {
	char preOrder[30];
	char inOrder[30];
	while (scanf("%s%s", preOrder, inOrder) != EOF) {
		TreeNode* proot;
		proot = Rebuild(preOrder, inOrder);
		postOrder(proot);
		printf("\n");
	}
	return 0;
}