LeetCode105 Construct Binary Tree from Preorder and Inorder Traversal

Given preorder and inorder traversal of a tree, construct the binary tree. （Medium）

Note:
You may assume that duplicates do not exist in the tree.

 

分析：

首先考虑已知前序遍历和中序遍历的情况下，手算恢复二叉树的方法。

比如序列层序遍历为1, 23, 4567的二叉树

其前序遍历：1,2,4,5,3,6,7

中序遍历为：4,2,5,1,6,3,7

前序遍历的方式是根 左 右，所以前序遍历的第一个一定是整棵树的根节点，在中序遍历中找到这个点，即可确定左右子树的中序遍历的范围。

即1为根， 2,4,5为左子树的前序遍历, 4,2,5为左子树的中序遍历，右子树同理。

这样递归的对2,4,5的前序遍历和4,2,5的中序遍历求解，只到空节点为止，即可建立起二叉树。

代码：

首先写的时候每次递归都新建了一个vector，导致了超内存，后来改成用辅助函数添加起始终止位置即可。

两份代码都记录如下

//代码1，临时vector过多而超内存
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int sz = preorder.size();
        if (sz == 0) {
            return nullptr;
        }
        int rootVal = preorder[0];
        TreeNode* root = new TreeNode(rootVal);
        int pos;
        for (pos = 0; pos < sz; ++pos) {
            if (inorder[pos] == rootVal) {
                break;
            }
        }
        if (pos < 1) {
            root -> left = nullptr;
        }
        else {
            vector<int> v1 = vector<int>(preorder.begin() + 1, preorder.begin() + pos + 1);
            vector<int> v2 = vector<int>(inorder.begin(), inorder.begin() + pos );
            cout << v1.size() << " " << endl;
            root -> left = buildTree(v1, v2);            
        }
        if (pos + 1 >= preorder.size()) {
            root -> right = nullptr;
        }
        else {
            vector<int> v3 = vector<int>(preorder.begin() + pos + 1, preorder.end());
            vector<int> v4 = vector<int>(inorder.begin() + pos + 1, inorder.end());
            root -> right = buildTree(v3, v4);  
        }
        return root;
    }
};

 

//代码2，改为利用记录起始终止位置的辅助函数即可
class Solution {
private:
    TreeNode* helper(vector<int>& preorder, int prevStart, int prevEnd, vector<int>& inorder, int inStart, int inEnd) {
        int sz = prevEnd - prevStart;
        if (sz == 0) {
            return nullptr;
        }
        int rootVal = preorder[prevStart];
        TreeNode* root = new TreeNode(rootVal);
        int pos;
        for (pos = 0; pos < sz; ++pos) {
            if (inorder[inStart + pos] == rootVal) {
                break;
            }
        }
        if (pos < 0) {
            root -> left = nullptr;
        }
        else {
            root -> left = helper(preorder, prevStart + 1, prevStart + pos + 1, inorder, inStart, inStart + pos);            
        }
        if (pos + 1 >= sz) {
            root -> right = nullptr;
        }
        else {
            root -> right = helper(preorder, prevStart + pos + 1, prevEnd, inorder, inStart + pos + 1, inEnd);  
        }
        return root;
    }
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
       return helper(preorder, 0, preorder.size(), inorder, 0, inorder.size());
    }
};