剑指offer T4 重建二叉树

 二叉树的三种遍历：

前序：根节点 --> 左子树 --> 右子树

中序：左子树 --> 根节点 --> 右子树

后序：左子树 --> 右子树 --> 根节点

在前序遍历序列中，从左到右依次取一个数，并找到其在中序遍历中的位置，左边为左子树，右边则是右子树

 按照这种思路，编写代码如下：

TreeNode* helper(vector<int> pre, vector<int> vin, int& k, int l, int r)　　
//k表示当前是前序序列中的元素的下标，l，r表示当前中序序列中的左子树或者右子树的边界
    {
        if (k == pre.size())
            return nullptr;
        if (l == r)
            return new TreeNode(vin[l]);
        int i = l;
        for (; i <= r; i++)　　//找到根节点的位置
        {
            if (vin[i] == pre[k]) 
                break;
        }
        TreeNode* root = new TreeNode(pre[k]);

        if (i > l)
            root->left = helper(pre, vin, ++k, l, i - 1);　　//进入左子树递归
        if (i < r)
            root->right = helper(pre, vin, ++k, i + 1, r);

        return root;

    }
    
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if (vin.size() != pre.size() || pre.size() == 0)
            return nullptr;
        int k = 0;
        TreeNode* head = helper(pre, vin, k, 0, vin.size() - 1);
        return head;
    }

.

 这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 　　这里是分界线 

类似的一道题

 前序遍历是从左往右依次是根节点，左子树，右子树；而后序遍历是从右往左依次是根节点，右子树，左子树。

所以上述代码只要稍作修改就行了。

TreeNode* helper(vector<int>& post, vector<int>& vin, int& k, int l, int r)
    {
        if (k < 0 )
            return nullptr;
        if (l == r)
            return new TreeNode(vin[l]);
        int i = r;
        for (; i >= l; i--)
        {
            if (vin[i] == post[k]) 
                break;
        }
        TreeNode* root = new TreeNode(post[k]);

        if (i < r)
            root->right = helper(post, vin, --k, i + 1, r);
        if (i > l)
            root->left = helper(post, vin, --k, l, i - 1);

        return root;

    }
    
    TreeNode* reConstructBinaryTree(vector<int>& in, vector<int>& post)
    {
        if (in.size() != post.size() || post.size() == 0)
            return nullptr;
        int k = post.size()-1;
        TreeNode* head = helper(post, in, k, 0, in.size() - 1);
        return head;
    }