struct BinaryTreeNode
{
int value;
BinaryTreeNode *Left;
BinaryTreeNode *Right;
};
BinaryTreeNode *ConstructCore(int *startPreorder, int *endPreorder, int *startInorder, int *endInorder)
{
//创建一个结节点
int rootValue = startPreorder[0];
BinaryTreeNode* root = new BinaryTreeNode();
root->value = rootValue;
root->Left = root->Right = nullptr;
if (startPreorder == endPreorder)
{
if (startInorder == endInorder && *startPreorder == *startInorder)
return root;
else
throw std::exception("Invalid input.");
}
//在中序遍历中找到根结点
int *rootInorder = startInorder;
while (startInorder <= endInorder && *rootInorder != rootValue)
++rootInorder;
if (rootInorder == endInorder && *rootInorder != rootValue)
throw std::exception("Invalid Input.");
int leftLength = rootInorder - startPreorder;
int* leftPreorderEnd = startPreorder + leftLength;
if (leftLength > 0)
{
//构建左子树
root->Left = ConstructCore(startPreorder + 1, leftPreorderEnd, startInorder, rootInorder - 1);
}
//构建右子树
if (leftLength < endPreorder-startPreorder)
{
root->Right = ConstructCore(leftPreorderEnd+1,endPreorder,rootInorder+1,endInorder);
}
return root;
}
BinaryTreeNode *Construct(int *preorder, int *inorder, int length)
{
if (preorder == nullptr || inorder == nullptr || length <= 0)
return nullptr;
return ConstructCore(preorder, preorder + length - 1, inorder, inorder + length - 1);
}
二叉树-----已知先序遍历、中序遍历,重建二叉树
猜你喜欢
转载自blog.csdn.net/weixin_39916039/article/details/82148701
今日推荐
周排行