题目:给定一颗树的先序遍历结果和中序遍历的结果,重建这颗树。
例如:
前序:a b d c e f
中序:d b a e c f
a是根节点。
思路:根据前序,可以很快找出根节点a,然后根据中序,可以找出根节点a的左右子树,然后递归求a的左子树前序b d跟中序d b,a的右子树前序c e f跟中序e c f即可
代码:
struct pNode {
pNode *pLeft;
pNode *pRight;
char ch;
};
void reBuild(char *pPreOrder, char *pInOrder, int treeLength, pNode **pRoot) {
if(pPreOrder == NULL || pInOrder == NULL) return ; //检查边界条件
pNode *pTemp = new pNode; //获得前序遍历的第一个节点
pTemp->pLeft = NULL;
pTemp->pRight = NULL;
pTemp->ch = *pPreOrder;
if(*pRoot == NULL) *pRoot = pTemp; //如果节点为空,把当前节点复制到根节点
if(treeLength == 1) return ; //如果当前树的长度是1,那么已经是最后一个节点了
char *pLeftEnd = pInOrder; //用来寻找左子树的长度
int nTempLen = 0;
while(*pLeftEnd != *pPreOrder) { //利用中序遍历的结果来找到左子树的结尾
if(pLeftEnd == NULL || pPreOrder == NULL) return ;
nTempLen++;
if(nTempLen > treeLength) break; //记录临时长度,以免溢出
pLeftEnd++;
}
int nLeftLen = (int)(pLeftEnd - pInOrder); //记录左子树的长度
int nRightLen = treeLength - nLeftLen - 1; //记录右子树的长度
if(nLeftLen > 0) { //重建左子树
reBuild(pPreOrder+1, pInOrder, nLeftLen, &((*pRoot)->pLeft));
}
if(nRightLen > 0) { //重建右子树
reBuild(pPreOrder+nLeftLen+1, pInOrder+nLeftLen+1, nRightLen, &((*pRoot)->pRight));
}
}