【剑指Offer】重建二叉树

题目描述

请实现一个函数，将一个字符串中的空格替换成“%20”。例如，当字符串为We Are Happy.则经过替换之后的字符串为We%20Are%20Happy。

解题思路

输入的是前序遍历及中序遍历的结果，如示例：

前序 : 1 2 4 7 3 5 6 8

中序 : 4 7 2 1 5 3 8 6

前序遍历的第一个数肯定为root，然后分别构建它的左右子树

在中序遍历里找到它的前一个数，若无，则左子树为NULL；

若有，则构建左子树；再找到它的后一个数，

若无，则右子树为NULL；若有，则构建右子树；

然后以root所在中序遍历数组位置分割成两半，记录前一半，

即左子树的大小，然后在前序遍历中也把左子树和右子树的子串

提出来，分别作为两棵树，重建，直到提出来的子串长度为0.

1

2 3

4 5 6

7 8

代码实现

class Solution7 {
public:
    struct TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> in) {
        return rebulidBinaryTree(pre, in);
    }

    static TreeNode* rebulidBinaryTree(vector<int>pre, vector<int>in){
        if (pre.size() == 0 || in.size() == 0){
            return NULL;
        }
        //获取根节点
        TreeNode* root = new TreeNode(pre[0]);
        //查找根节点在中序遍历中的位置
        int cur = -1;
        for (int i = 0; i < in.size(); i++){
            if (pre[0] == in[i]){
                cur = i;
                break;
            }
        }
        if (cur == -1){
            return NULL;
        }
        //构建左子树
        vector<int> lin;
        for (int i = 0; i < cur; i++){
            lin.push_back(in[i]);
        }
        vector<int> lpre;
        for (int i = 1; i < lin.size() + 1; i++){
            lpre.push_back(pre[i]);
        }

        root->left = rebulidBinaryTree(lpre, lin);

        //构建右子树
        vector<int> rin;
        for (int i = cur+1; i < in.size(); i++){
            rin.push_back(in[i]);
        }
        vector<int> rpre;
        for (int i = 1+lin.size(); i < pre.size() ; i++){
            rpre.push_back(pre[i]);
        }

        root->right = rebulidBinaryTree(rpre, rin);
        return root;

    }
};