剑指offer—重构二叉树

剑指offer—重构二叉树

1.题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

---

结构体：

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

---

所以前序遍历序列是{1, 2, 4, 7, 3, 5, 6, 8};中序遍历序列是{4, 7, 2, 1, 5, 3, 8, 6};

2.思路

1. 因为结构是二叉树，而二叉树本身就是一个递归定义，所以使用递归思想。
2. 前序遍历序列中第一个数字1就是根节点，在中序遍历序列中，数字1的左边3个数字4, 7, 2是根节点的左子树，数字1的右边四个数字5, 3, 8, 6是根节点的右子树。在前序遍历中，从根节点数字1之后的3个数字2, 4, 7是根节点的左子树，再之后的4个数字3, 5, 6, 8是根节点的右子树。这样，就找到了左右子树对应的子序列。
3. 之后使用递归的方法完成

3.代码

TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
    int sizePre = pre.size();
    int sizeVin = vin.size();
    if(sizePre != sizeVin || 0 == sizePre)
        return NULL;
    return reConstructBinaryTreeCore(pre,vin,0,sizePre - 1,0,sizePre - 1);
}
TreeNode* reConstructBinaryTreeCore(vector<int>& pre,vector<int>& vin,
                                    int preBegin, int preEnd, int vinBegin, int vinEnd)
{
    int value = pre[preBegin];
    TreeNode* root = new TreeNode(value);
    //rootIndex vin根节点的下标
    int rootIndex = vinBegin;
    while(rootIndex <= vinEnd && value != vin[rootIndex]) ++rootIndex;
    //中序遍历中rootIndex左边元素的长度
    int leftLen = rootIndex - vinBegin;
    //中序遍历中rootIndex右边元素的长度
    int rightLen = vinEnd - rootIndex;

    if(leftLen > 0)
        root->left = reConstructBinaryTreeCore(pre, vin, preBegin + 1,preBegin + leftLen,
                                               vinBegin,rootIndex - 1);
    if(rightLen > 0)
        root->right = reConstructBinaryTreeCore(pre, vin, preBegin + leftLen + 1,preEnd,
                                                rootIndex + 1,vinEnd);
    return root;
}