leetcode 106. 从中序与后序遍历序列构造二叉树

leetcode 106. 从中序与后序遍历序列构造二叉树

题目详情

题目链接
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

• 例如，给出
  中序遍历 inorder = [9,3,15,20,7]
  后序遍历 postorder = [9,15,7,20,3]
  返回如下的二叉树：
  3
  / \
  9 20
  / \
  15 7

我的代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* getTree(vector<int>& inorder, int start1, int stop1, vector<int>& postorder, int start2, int stop2) {
        if (start1 > stop1 || start2 > stop2) {
            return NULL;
        }
        int rootValue = postorder[stop2];
        auto root = new TreeNode(rootValue);
        int pos = find(inorder.begin() + start1, inorder.end(), rootValue) - inorder.begin() - start1;
        root->left = getTree(inorder, start1, start1 + pos - 1, postorder, start2, start2 + pos - 1);
        root->right = getTree(inorder, start1 + pos + 1, stop1, postorder, start2 + pos, stop2 - 1);
        return root;
    }

    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        return getTree(inorder, 0, inorder.size() - 1, postorder, 0, postorder.size() - 1);
    }
};

我的成绩

执行结果：通过
执行用时：60 ms, 在所有 C++ 提交中击败了20.31%的用户
内存消耗：17.2 MB, 在所有 C++ 提交中击败了59.78%的用户

一些想法

通过递归循环构建二叉树。

执行用时为 4 ms 的范例

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    private:
        unordered_map <int,int> inorder_index;
    public:
        TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
            int n=inorder.size();
            for(int i=0;i<n;++i){
                inorder_index[inorder[i]]=i;
            }
            return mytree(inorder, postorder, 0,n-1,0,n-1);
        }

        TreeNode* mytree(const vector<int>& inorder,const vector<int>& postorder, int inorder_left, int                                    inorder_right, int postorder_left, int postorder_right) {
        if (inorder_left > inorder_right) return nullptr;
        int inorder_rootnum = inorder_index[postorder[postorder_right]];
        int left_length = inorder_rootnum - inorder_left; //
        int right_length = inorder.size() - left_length - 1;
        TreeNode* root = new TreeNode(postorder[postorder_right]);
        root->left = mytree(inorder, postorder, inorder_left, inorder_left + left_length - 1, postorder_left,                                          postorder_left + left_length - 1);
        root->right = mytree(inorder, postorder, inorder_left + left_length + 1, inorder_right, postorder_left +                                       left_length, postorder_right - 1);
        return root;
    }
};

思考

参见官方解答