~由前序遍历和中序遍历重建二叉树~

二叉树经典面试题之一：由前序遍历和中序遍历重建二叉树

                                        如：前序序列：1 2 3 4 5 6

                                               中序序列：3 2 4 1 6 5

 

解题思路：

二叉树的前序遍历规则：遍历根结点；前序遍历左子树，前序遍历右子树

              中序遍历规则：中序遍历左子树，遍历根结点，中序遍历右子树

以题中所给的序列为准，由遍历规则得：节点1为根结点，所以可以把中序序列分为三个部分：左子树 3 2 4，根节点1，右子树 6 5。

同理，中序序列左子树部分的2为根结点、3为左子树、4为右子树；

           中序序列右子树部分的5为根结点、6为左子树。

 

完整的源代码及测试用例如下：

BinaryTreeNode<int>* Rebuild(int* Prev, int prev_start, int prev_end, int* In, int in_start, int in_end)
{
	if(Prev == NULL || In == NULL)
	{
		return NULL;
	}

	//序列长度必须相等
	if(prev_end - prev_start != in_end - in_start)
	{
		return NULL;
	}

	BinaryTreeNode<int>* root = new BinaryTreeNode<int>(Prev[prev_start]);
	root->_left = NULL;
	root->_right = NULL;

	if(prev_end == prev_start)
	{
		return root;
	}

	int index = 0;
	int len = 0;
	for(index = in_start; index <= in_end; ++index)
	{
		if(Prev[prev_start] == In[index])
		{
			break;
		}
	}

	//存在左子树
	if(index > in_start)
	{
		root->_left = Rebuild(Prev, prev_start+1, prev_start+index, In, in_start, in_start+index - 1);
	}

	//存在右子树
	if(index < in_end)
	{
		root->_right = Rebuild(Prev, prev_start+index+1, prev_end, In, in_start+index+1, in_end);
	}

	return root;
}

void TestRebuild()
{
	int Prev[] = {1, 2, 3, 4, 5, 6};
	int prev_start = 0;
	int prev_end = sizeof(Prev) / sizeof(Prev[0]) - 1;
		
	int In[] = {3, 2, 4, 1, 6, 5};
	int in_start = 0;
	int in_end = sizeof(In) / sizeof(In[0]) - 1;

	BinaryTreeNode<int>* root = Rebuild(Prev, prev_start, prev_end, In, in_start, in_end);
}