由双遍历序列构造二叉树-河南理工大学-CSDN博客

本文链接：https://blog.csdn.net/m0_72957271/article/details/134040491

任务描述

本关任务：实现 ConstructTree.cpp 里的TNode* InPreToTree(char *pa, char *ia, int p1, int p2, int i1, int i2)函数。

编程要求

本关任务是实现 ConstructTree.cpp 里的TNode* InPreToTree(char *pa, char *ia, int p1, int p2, int i1, int i2)函数。

该函数的功能是由前序遍历序列和中序遍历序列构造二叉树。前序序列为pa[p1:p2]，中序序列为ia[i1:i2]，返回所构造的二叉树的根指针。

提示1：这是一个递归函数，在主程序中调用： InPreToTree(pa,ia,0,n-1,0,n-1)，其中n是序列长度。 提示2：由于在DeleteTree()中是使用delete删除一个树结点，所以在InPreToTree()需要使用new来申请结点空间。

//ConstructTree.cpp
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "ConstructTree.h"
/*
InPreToTree(): 由前序遍历序列和中序遍历序列构造二叉树
前序序列为pa[p1:p2]
中序序列为ia[i1:i2]
返回所构造的二叉树的根指针
*/
TNode* InPreToTree(char *pa, char *ia,
int p1, int p2, int i1, int i2)
{
//在begin和end之间添加你的代码
/********* begin **********/
/********* end ************/
}
void PrintPostTravel(TNode* t)
{
if(t==NULL) return;
if(t->left) PrintPostTravel(t->left);
if(t->right) PrintPostTravel(t->right);
printf("%c", t->data);
}
void DeleteTree(TNode* t)
{
if(t==NULL) return;
if(t->left) DeleteTree(t->left);
if(t->right) DeleteTree(t->right);
delete t;
}

测试说明

本关的测试过程如下：

平台编译 step1/Main.cpp ;
平台运行该可执行文件，并以标准输入方式提供测试输入；
平台获取该可执行文件的输出，然后将其与预期输出对比，如果一致则测试通过；否则测试失败。

输入格式：输入前序序列输入中序序列

输出格式：输出后序序列

以下是平台对 step1/Main.cpp 的测试样例：

样例输入 ABDECFG DBEAFCG

样例输出 Post Travel Result：DEBFGCA

#define _CRT_SECURE_NO_WARNINGS 1

///
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "ConstructTree.h"
/

/*
InPreToTree(): 由前序遍历序列和中序遍历序列构造二叉树
前序序列为pa[p1:p2]
中序序列为ia[i1:i2]
返回所构造的二叉树的根指针
*/


TNode* InPreToTree(char* pa, char* ia, int p1, int p2, int i1, int i2)
{
	//pa指向前序序列的字符数组的首地址
	//ia指向中序序列的字符数组的首地址
	//p1和p2是pa前序序列的首元素下标和最后一个元素下标
	//i1和i2是pa中序序列的首元素下标和最后一个元素下标


	/*这里注意不要使用p1 >= p2，当p1 == p2时，表明是遇到了叶子结点，依然可以进行，
	如果使用 == 了，最后的叶子结点不会加入树中，会出现错误*/


	if ((p1 > p2) || (i1 > i2)) return NULL; //就是左子树和右子树都只剩下一个元素的时候，停止递归


	TNode* T = (TNode*)malloc(sizeof(TNode));
	T->data = pa[p1];   //pa[p1]是前序序列的首元素

	
	int Llen = 0;  //用来记录序列中左子树的长度
	int i = i1;
	while (ia[i] != T->data)
	{//前序首元素就是根结点，一直循环，直到在中序中找到根节点，此时，Llen就是左子树长度
		i++;
		Llen++;
	}

	if (Llen == 0)   //左子树为0的情况，如前序序列是ABCD,中序序列是ACDB的情况
	{
		T->right = InPreToTree(pa, ia, p1 + 1, p2, i1 + 1, i2);
		T->left = NULL;
	}


	/*不可以只靠最大的那颗树来思考判断条件，最大的树，首个元素坐标为0，但子树的首元素坐标
	不是重新设置为0了，而是依然在原数组中的位置*/
	else if (Llen == i2 - i1)  //右子树为0,不是Llen == i2,是用最后一个元素坐标减去首元素坐标。
	{
		T->left = InPreToTree(pa, ia, p1 + 1, p2, i1, i2 - 1);
		T->right = NULL;
	}

	else
	{

		T->left = InPreToTree(pa, ia, p1 + 1, p1 + Llen, i1, i1 + Llen - 1);

		// T->left = InPreToTree(pa, ia, p1 + 1, p1 + Llen, i1, i2 - Llen - 1);  这是错误的
		/*在输入为ABDECFG和DBEAFCG时，i1 + Llen - 1  和  i2 - Llen - 1  两种计算方法在巧合下显示一致
		但如输入的是ABCDEF和BADEFC会发现，只有i1 + Llen - 1 是正确的 */

		T->right = InPreToTree(pa, ia, p1 + Llen + 1, p2, i1 + Llen + 1, i2);
	}

	return T;

}



void PrintPostTravel(TNode* t)
{
	if (t == NULL) return;
	if (t->left) PrintPostTravel(t->left);
	if (t->right) PrintPostTravel(t->right);
	printf("%c", t->data);
}

void DeleteTree(TNode* t)
{
	if (t == NULL) return;
	if (t->left) DeleteTree(t->left);
	if (t->right) DeleteTree(t->right);
	delete t;
}


int main()
{
	char pre[100], in[100];

	int n; //用来保存序列长度

	TNode* t;//用来保存二叉树的根

	scanf("%s", pre);
	scanf("%s", in);

	n = 0; 
	while (pre[n]) //计算序列的长度
	{
		n++;
	}

	t = InPreToTree(pre, in, 0, n - 1, 0, n - 1);

	printf("Post Travel Result：");
	PrintPostTravel(t);

	printf("\n");

	DeleteTree(t);
	return 0;
}

在头歌运行时请删除主函数