【编程之美】3.9重建二叉树

题目：

给定一棵二叉树，假定每个节点都用唯一的字符表示，具体结构如下：

struct NODE {
NODE* pLeft;
NODE* pRight;
char chValue;
};
假设已经有了前序遍历和中序遍历结果，希望通过一个算法重建这棵树。
给定函数的定义如下：

void Rebuild(char* pPreOrder, char* pInOrder, int nTreeLen, NODE** pRoot);
参数
pPreOrder:前序遍历结果的字符串数组。
pInOrder: 中序遍历结果的字符串数组。
nTreeLen: 树的长度。
pRoot: 根据前序和中序遍历结果重新构建树的根节点。
例如
前序遍历结果：a b d c e f
中序遍历结果：d b a e c f

思路：不难，根据前序遍历结果找跟，根据中序遍历结果划分左右子树，递归求解。

/*
start time = 15:55
end time = 16:23
*/
#include<iostream>
using namespace std;

typedef struct NODE{
NODE *pLeft, *pRight;
char chValue;
}NODE;

void Rebuild(char* pPreOrder, char* pInOrder, int nTreeLen, NODE** pRoot)
{
if(pPreOrder == NULL || pInOrder == NULL) //边界条件不要忘了 这是看答案后补上的
{
return;
}
if(nTreeLen <= 0)
{
return;
}
//重建根节点 一定是先序遍历的第一个值
*pRoot = new NODE;
(*pRoot)->chValue = pPreOrder[0];
(*pRoot)->pLeft = NULL;
(*pRoot)->pRight = NULL;

//查找根在中序遍历中的位置，把中序遍历表转化为左子树和右子树两个部分
int iLocateRoot;
for(iLocateRoot = 0; iLocateRoot < nTreeLen; iLocateRoot++)
{
if(pInOrder[iLocateRoot] == pPreOrder[0])
{
break;
}
}

//重建左子树
Rebuild(pPreOrder+1, pInOrder, iLocateRoot, &((*pRoot)->pLeft));
//重建右子树
Rebuild(pPreOrder+iLocateRoot+1, pInOrder + iLocateRoot + 1, nTreeLen - iLocateRoot - 1, &((*pRoot)->pRight));
}

int main()
{
char pre[7] = {'a','b','d','c','e','f','\0'};
char in[7] = {'d','b','a','e','c','f','\0'};
NODE * T = NULL;
Rebuild(pre, in, 7, &T);

return 0;
}