面试题6：重建二叉树

/*面试题6
  题目：输入某二叉树的前序遍历和中序遍历的结果，重建出该二叉树。不包含重复数字。
  前序遍历的第一个结果就是根节点，中序遍历中根节点前面的节点就是左子树，后面的节点就是右子树。然后递归的构建左右子树。
*/
BinaryTreeNode* ConstructBinaryNode(int* startPreorder,int* endPreorder,int* startInorder,int* endInorder);
BinaryTreeNode* Construct(int* preorder,int* inorder,int length)
{
if(preorder==NULL||inorder==NULL||length<=0)
return NULL;
return ConstructBinaryNode(preorder,preorder+length-1,inorder,inorder+length-1);
}
BinaryTreeNode* ConstructBinaryNode(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=NULL;

if(startPreorder==endPreorder)
{
if(startInorder==startInorder&&*startPreorder==*startInorder)
return root;//只有一个节点的情况
else
throw exception("invalid input.");
}
//在中序遍历中查找根节点的值
int* rootInorder=startInorder;
while(rootInorder<=endInorder&&*rootInorder!=rootValue)
++rootInorder; //找到根节点在中序中的位置
if(rootInorder==endInorder&&*rootInorder!=rootValue)//没有找到根节点抛出异常
throw exception("invalid input.");
int leftLength=rootInorder-startInorder;//左子树的长度
int* leftPreorderEnd=startPreorder+leftLength;//前序中左子树的起点
if(leftLength>0)
root->m_pLeft=ConstructBinaryNode(startPreorder+1,leftPreorderEnd,startInorder,rootInorder-1);//构建左子树
if(leftLength<endPreorder-startPreorder)
root->m_pRight=ConstructBinaryNode(leftPreorderEnd+1,endPreorder,rootInorder+1,endInorder);//构建右子树
return root;
}
/*测试用例包括:完全二叉树，不完全二叉树，只有左子节点的二叉树，只有右子节点的二叉树，只有根节点的二叉树，根节点为NULL，输入的前序和中序不匹配。*/