- 题目:根据某二叉树的前序中序序列重建二叉树,假设前序中序序列中不包含重复的数字。树的结点定义如下。
struct BinaryTreeNode
{
BinaryTreeNode(char data)
:_pLeft(NULL)
, _pRight(NULL)
, _data(data)
{}
BinaryTreeNode *_pLeft;
BinaryTreeNode *_pRight;
char _data;
};
思路:前序序列的第一个值就是根节点的值,在中序序列中,根结点的值在序列的中间,左子树的值位于根节点值的左边,右子树节点的值位于根节点的值的右边,根据此思路,遍历中序序列,找到根结点的值,划分出左右子树,递归创建出二叉树。
- 实现代码
BinaryTreeNode *ConstructCore(char *preStart, char *preEnd, char *inStart, char *inEnd);
BinaryTreeNode *Construct(char *preStart, char *inStart, int length)
{
if (NULL == preStart || NULL == inStart || length <= 0)
return NULL;
return ConstructCore(preStart,preStart + length - 1,inStart,inStart + length - 1);
}
BinaryTreeNode *ConstructCore(char *preStart, char *preEnd, char *inStart, char *inEnd)
{
char rootvalue = preStart[0];//前序遍历的第一个节点是树的根
BinaryTreeNode * root = new BinaryTreeNode(rootvalue);
if (preStart == preEnd)
{
if (inStart == inEnd && *inStart == rootvalue)
return root;
else
{
cout << "输入的序列有误" << endl;
return NULL;
}
}
//在中序遍历序列中寻找根所在的位置
char *inRoot = inStart;
while (inRoot <= inEnd && *inRoot != rootvalue)
++inRoot;
if (inRoot == inEnd && *inRoot != rootvalue)
{
cout << "输入的序列有误" << endl;
return NULL;
}
int leftsize = inRoot - inStart;
int rightsize = inEnd - inRoot;
if (leftsize > 0)
{
root->_pLeft = ConstructCore(preStart + 1, preStart + leftsize, inStart, inRoot - 1);
}
if (rightsize > 0)
{
root->_pRight = ConstructCore(preStart + leftsize + 1,preEnd,inRoot + 1,inEnd);
}
return root;
}