标签:
题目:给定一颗树的先序遍历结果和中序遍历的结果,重建这颗树。
例如:
前序: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)); } }
标签:
原文地址:http://blog.csdn.net/u010470972/article/details/45558315