九度OJ-剑指offer-题目1385：重建二叉树

最新推荐文章于 2019-06-14 17:25:17 发布

原创最新推荐文章于 2019-06-14 17:25:17 发布 · 830 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer 重建二叉树

本文详细介绍了如何通过前序遍历和中序遍历序列构建二叉树，并阐述了如何从构建的二叉树中获取后序遍历序列的过程。文章深入探讨了二叉树遍历的相关概念，旨在帮助读者理解二叉树结构及其遍历算法的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并输出它的后序遍历序列。

输入：

输入可能包含多个测试样例，对于每个测试案例，

输入的第一行为一个整数n(1<=n<=1000)：代表二叉树的节点个数。

输入的第二行包括n个整数(其中每个元素a的范围为(1<=a<=1000))：代表二叉树的前序遍历序列。

输入的第三行包括n个整数(其中每个元素a的范围为(1<=a<=1000))：代表二叉树的中序遍历序列。

输出：

对应每个测试案例，输出一行：

如果题目中所给的前序和中序遍历序列能构成一棵二叉树，则输出n个整数，代表二叉树的后序遍历序列，每个元素后面都有空格。

如果题目中所给的前序和中序遍历序列不能构成一棵二叉树，则输出”No”。

样例输入：

81 2 4 7 3 5 6 84 7 2 1 5 3 8 681 2 4 7 3 5 6 84 1 2 7 5 3 8 6

样例输出：

7 4 2 5 8 6 3 1 No

#include <iostream>
#include <stdio.h>
using namespace std;

struct BinaryTreeNode
{
	int data;
	BinaryTreeNode *lchild;
	BinaryTreeNode *rchild;
};

BinaryTreeNode* ConstructCore(
							  int* startPreorder, int* endPreorder,
							  int* startInorder, int* endInorder
							  )
{
	int rootvalue = startPreorder[0];
	BinaryTreeNode* root = new BinaryTreeNode;
	root ->data = rootvalue;
	root ->lchild = root ->rchild = NULL;
	if(startPreorder == endPreorder)
	{
		if(startInorder == endInorder && *startPreorder == *startInorder)
			return root;
		else
			return NULL;
	}
	int* rootInorder = startInorder;
	while(rootInorder <= endInorder && *rootInorder != rootvalue)
	{
		++rootInorder;
	}
	if(rootInorder == endInorder && *rootInorder != rootvalue)
	{
		return NULL;
	}
	int leftLength = rootInorder - startInorder;
	int* leftPreorderEnd = startPreorder + leftLength;
	if(leftLength > 0)
		root ->lchild = ConstructCore(startPreorder + 1, leftPreorderEnd, startInorder, rootInorder -1);
	if(leftLength < endPreorder - startPreorder)
		root ->rchild = ConstructCore(leftPreorderEnd +1, endPreorder, rootInorder + 1, endInorder);
	return root;
}

BinaryTreeNode* Construct(int* preorder, int* inorder, int length)
{
	if(preorder == NULL || inorder == NULL || length <= 0)
		return NULL;
	return ConstructCore(preorder, preorder + length - 1, inorder, inorder + length - 1);
}

void PostOrderSum(BinaryTreeNode *p, int *list, int &sum)
{
	if(p != NULL)
	{
		PostOrderSum(p ->lchild, list, sum);
		PostOrderSum(p ->rchild, list, sum);
		list[sum ++] = p ->data;
	}
}

int main()
{
	int length;
	while(scanf("%d", &length) != EOF)
	{
		int *preorder = new int[length];
		int *inorder = new int[length];
		int i = 0; 
		for(i = 0; i < length; i++)
			scanf("%d", &preorder[i]);
		for(i = 0; i < length; i++)
			scanf("%d", &inorder[i]);
		BinaryTreeNode *p = Construct(preorder, inorder, length);
		int *postorder = new int[length];
		int sum = 0;
		PostOrderSum(p, postorder, sum);
		if(sum == length)
		{
			for(i = 0; i < length; i++)
			{
					printf("%d ", postorder[i]);
			}
			printf("\n");
		}
		else
			printf("No\n");
	}
	return 0;
}