题干:
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
限制:
0 <= 节点个数 <= 5000
解题思路:
需要理解前序遍历、中序遍历、后序遍历的排列方法。通过前序遍历头节点一定是根节点,搜索中序遍历中根节点的位置,可知,前序遍历可以分为:
根节点——左子树的值——右子树的值
中序遍历可以分为:
左子树的值——根节点——右子树的值
根据子树特点,我们可以通过同样的方法对左(右)子树进行划分,每轮可确认三个节点的关系 。此递推性质让我们联想到用 递归方法 处理。
代码实现:
struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize){
//截止条件
if (preorderSize == NULL || inorderSize == NULL){
return NULL;
}
if (preorderSize == 0 || inorderSize == 0){
return NULL;
}
//初始化
struct TreeNode* node = (struct TreeNode*)malloc(sizeof(struct TreeNode));
node -> val = preorder[0];
node -> left = NULL;
node -> right = NULL;
int pindex;
for (pindex = 0; pindex < inorderSize; pindex++){
if (inorder[pindex] == preorder[0]){
break;
}
}
node -> left = buildTree(preorder + 1, pindex, inorder, pindex);
node -> right = buildTree(preorder + pindex + 1, preorderSize - pindex - 1, inorder + pindex + 1, inorderSize - pindex - 1);
return node;
}