《剑指offer》面试题7：重建二叉树

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

 解析：

前序遍历 --> 确定 第一个数字为根结点 

根结点的值  & 中序遍历 --> 1 前面是左子树、后是右子树

中序遍历左子树三个 --> 前序遍历1后前3个位为左子树，剩余为右

 

定义头文件  BrinaryTree.h

#pragma once  //保证：同一个文件不会被包含多次

// 二叉树的链式存储
// 结点类型定义
struct BinaryTreeNode
{
	int              m_nValue;
	BinaryTreeNode*  m_pLeft;
	BinaryTreeNode*  m_pRight;
};

递归完成遍历：

#include "..\treeCode\BinaryTree.h"
#include <exception>
#include <cstdio>

BinaryTreeNode* ConstructCore(int* startPreorder, int* endPreorder, int* startInorder, int* endInorder);
// 参数
// startPreorder:前序遍历的字符串数组的首
// endPreorder： 前序遍历的字符串数组的尾
// startInorder：中序遍历的字符串数组的首
// endInorder：  中序遍历的字符串数组的尾

// 输入前序，中序遍历
BinaryTreeNode* Construct(int* preorder, int* inorder, int length)
{
	if (preorder == nullptr || inorder == nullptr || length <= 0)
		return nullptr;

	return ConstructCore(preorder, preorder + length - 1,inorder, inorder + length - 1);
}

BinaryTreeNode* ConstructCore(int* startPreorder, int* endPreorder, int* startInorder, int* endInorder)
{
	//前序遍历序列确定第一个数字是根结点的值
	int rootValue = startPreorder[0];
	BinaryTreeNode* root = new BinaryTreeNode();

	root->m_nValue = rootValue;
	root->m_pLeft = root->m_pRight = nullptr;

	if (startPreorder == endPreorder)  //？
	{
		if (startInorder == endInorder && *startPreorder == *startInorder)
			return root;
		else
			throw std::exception("Invalid input.");
	}

	// 在中序遍历中找到根结点的值
	int* rootInorder = startInorder;
	while (rootInorder <= endInorder && *rootInorder != rootValue) //遍历中序字符串数组
		++rootInorder;                                         //确定root位置

	if (rootInorder == endInorder && *rootInorder != rootValue)
		throw std::exception("Invalid input.");

	int leftLenght = rootInorder - startInorder;
	int* leftPreorderEnd = startPreorder + leftLenght;

	if (leftLenght > 0)
		// 构建左子树
		root->m_pLeft = ConstructCore(startPreorder + 1, leftPreorderEnd, startInorder, rootInorder - 1);
	if (leftLenght < endPreorder - startPreorder)
		//构建右子树
		root->m_pRight = ConstructCore(leftPreorderEnd + 1, endPreorder, rootInorder + 1, endInorder);

	return root;
}

//  --------------测试代码------------
void Test(char* testName, int* preorder, int* inorder, int length)
{
	if (testName != nullptr) // 打印测试名
		printf("%s begins:\n",testName);

	printf("preorder sequence: "); //打印前序遍历
	for (int i = 0; i < length; ++i)
		printf("%d", preorder[i]);
	printf("\n");

	try
	{
		BinaryTreeNode* root = Construct(preorder, inorder, length);
		PrintTree(root);

		DestroyTree(root);
	}
	catch (std::exception& exception)
	{
		printf("Invalid Input.\n");
	}

}


// 普通二叉树
//              1
//           /     \
//          2       3  
//         /       / \
//        4       5   6
//         \         /
//          7       8
void Test0()
{
	const int length = 8;
	int preorder[length] = { 1, 2, 4, 7, 3, 5, 6, 8 };
	int inorder[length] = { 4, 7, 2, 1, 5, 3, 8, 6 };

	Test("Test1", preorder, inorder, length);
}

int main(int argc, char* argv[])
{
	Test0();
	return 0;
}