根据前序和中序遍历结果构造二叉树

最新推荐文章于 2021-03-19 22:30:22 发布

待磨的咖啡豆

最新推荐文章于 2021-03-19 22:30:22 发布

阅读量547

点赞数

分类专栏：面试题目数据结构与算法文章标签： C++ 算法数据结构面试题

本文链接：https://blog.csdn.net/icandoit116/article/details/11356971

版权

面试题目同时被 2 个专栏收录

14 篇文章 0 订阅

订阅专栏

数据结构与算法

13 篇文章 0 订阅

订阅专栏

思路：前序第一个元素作为根，从中序找出左子树和右子树的序列，递归构建子树

TreeNode * 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);

}

TreeNode * ConstructCore(int* startPreOrder,int* endPreOrder,int* startInOrder,int* endInOrder)
{
	int rootValue=startPreOrder[0];
	TreeNode* root=new TreeNode();
	root->value=rootValue;
	root->lchild=root->rchild=NULL;

	if (startPreOrder==endPreOrder)
		if (startInOrder==endInOrder&&(*startPreOrder)==(*startInOrder))
		{
			return root;
		}
		else throw std::exception("error");

	//寻找根在中序中的位置
	int * rootInOrder=startInOrder;
	while(rootInOrder<=endInOrder&&*rootInOrder!=rootValue) rootInOrder++;
	if (rootInOrder==endInOrder&&*rootInOrder!=*endInOrder) throw std::exception("error");

	int leftLength=rootInOrder-startInOrder;
	int rightLength=endInOrder-rootInOrder;
	
	int* leftEndPreOrder=startPreOrder+leftLength;
	int* rightStartPreOrder=leftEndPreOrder+1;
	
	if (leftLength>0)
	{//构建左子树
		root->lchild=ConstructCore(startPreOrder+1,leftEndPreOrder,startInOrder,rootInOrder-1);
	}
	if (rightLength>0)
	{
		root->rchild=ConstructCore(rightStartPreOrder,endPreOrder,rootInOrder+1,endInOrder);
	}
	return root;
}