面试题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，输入的前序和中序不匹配。*/