剑指offer———重建二叉树

问题重述

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

思路解析：

解析本题的主要技巧就是要精确的找出顶节点，并找出左右两颗子树的所有元素，然后递归的进行二叉树的构建，具体的实现见代码部分。

代码实现：

struct binaryTreeNode
{
	int mData;
	binaryTreeNode * mLeft;
	binaryTreeNode * mRight;
};
//根据前序和中序序列重建二叉树
binaryTreeNode *createBt(int * pre, int * mid, int l1, int r1, int l2, int r2)
{
	if (l1 > r1 || l2 > r2)
	{
		return nullptr;
	}
        binaryTreeNode *s = new binaryTreeNode();
	s->mLeft = s->mRight = nullptr;
	int i = 0;
	for (i = l2; i < r2; ++i)
	{
		if (pre[l1] == mid[i])
		{
			break;
		}
	}
	s->mData = mid[i];
	//对左右子树元素的下标重新进行判断，重构二叉树
	s->mLeft = createBt(pre, mid, l1 + 1, l1 + i - l2, l2, i - 1);
	s->mRight = createBt(pre, mid, l1+i-l2+1,r1,i+1,r2);
	return s;
}



//二叉树的前序遍历和中序遍历，用于验证构建的二叉树是否正确
void preOrder(binaryTreeNode *s)
{
	if (s == nullptr)
	{
		return;
	}
	printf("%d ", s->mData);
	preOrder(s->mLeft);
	preOrder(s->mRight);
}

void inOrder(binaryTreeNode *s)
{
	if (s == nullptr)
	{
		return;
	}

	inOrder(s->mLeft);
	printf("%d ", s->mData);
	inOrder(s->mRight);
}

void destory(binaryTreeNode *s)
{
	if (s != nullptr)
	{
		binaryTreeNode *lChild = s->mLeft;
		binaryTreeNode *rChild = s->mRight;
		delete s;
		destory(lChild);
		destory(rChild);
	}
}

测试：

int main()
{
    int pre[] = { 1, 2, 4, 7, 3, 5, 6, 8 };
    int mid[] = { 4, 7, 2, 1, 5, 3, 8, 6 };
    binaryTreeNode *s = new binaryTreeNode();
    s = createBt(pre, mid, 0, 7, 0, 7);
    preOrder(s);
    printf("\n");
    inOrder(s);
    printf("\n");
    destory(s);
    return 0;
}
    

 

所得结果：将重建后的二叉树的先序后中序的遍历序列打印出来，发现和输入的一样，结果正确。