面试题7:重建二叉树

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

#include <iostream>
#include <string>
#include <cstdlib>

using namespace std;

typedef struct treenode
{
	int val;
	treenode* lchild;
	treenode* rchild;
}TreeNode, *Tree;

Tree constructTree(int* startPreOrder, int* endPreOrder, int* startInorder, int* endInorder)
{
	int rootVal = *startPreOrder;
	Tree root = new TreeNode;
	root->val = rootVal;
	root->lchild = nullptr;
	root->rchild = nullptr;

	if (startPreOrder == endPreOrder)
	{
		if (startInorder == endInorder && *startPreOrder == *startInorder)
			return root;
		else
			throw std::exception("Input Invalid");
	}
	int* rootInoder = startInorder;
	while (rootInoder <= endInorder && *rootInoder != rootVal)
		rootInoder++;
	if (rootInoder == endInorder && *rootInoder != rootVal)
		throw std::exception("Input Invaild");
	int leftLength = rootInoder - startInorder;
	int* leftEndProOrder = startPreOrder + leftLength;
	if (leftLength > 0)
	{
		root->lchild = constructTree(startPreOrder + 1, leftEndProOrder, startInorder, rootInoder - 1);
	}
	if (leftLength < endPreOrder - startPreOrder)
	{
		root->rchild = constructTree(leftEndProOrder + 1, endPreOrder, rootInoder + 1, endInorder);
	}
	return root;
}

int main()
{
	int preorder[8] = { 1, 2, 4, 7, 3, 5, 6, 8 };
	int inorder[8] = { 7, 4, 2,1, 5, 3, 8, 6 };
	Tree root = constructTree(preorder, preorder + 7, inorder, inorder + 7);
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值