LeetCode-105-从前序与中序遍历序列构造二叉树

1、递归

我们可以根据中序遍历和前序遍历的特点来构造二叉树。前序遍历的顺序是“根左右”，中序遍历的顺序是“左根右”，因此我们在前序序列中获得的第一位一定是根节点。我们可以将中序遍历的序列放入哈希表中，将值与位置进行映射，此时我们可以使用哈希查找获得中序遍历序列中根节点所在的位置，最终得到左子树和右子树的集合。我们记录下左子树的长度并构建根节点，而后利用长度递归查找左子树的根节点和右子树的根节点即可。

class Solution {
public:
    unordered_map<int, int> hash_inorder;

    TreeNode *findRoot(vector<int> &preorder, vector<int> &inorder, int preorder_low, int preorder_high, int inorder_low,
                    int inorder_high) {
        if (preorder_low > preorder_high) return nullptr;
        int inorder_root = hash_inorder[preorder[preorder_low]];
        TreeNode *root = new TreeNode(inorder[inorder_root]);
        int size_left_subtree = inorder_root - inorder_low;
        root->left = findRoot(preorder, inorder, preorder_low + 1, preorder_low + size_left_subtree, inorder_low,
                            inorder_root - 1);
        root->right = findRoot(preorder, inorder, preorder_low + size_left_subtree + 1, preorder_high, inorder_root + 1,
                            inorder_high);
        return root;
    }

    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        for (int i = 0; i < inorder.size(); ++i) {
            hash_inorder[inorder[i]] = i;
        }
        return findRoot(preorder, inorder, 0, preorder.size() - 1, 0, preorder.size() - 1);
    }
};

2、迭代

我们可以利用栈和指针辅助构造二叉树。初始时栈中只存放了根节点，即前序遍历中的第一个节点，指针指向中序遍历的第一个节点。此时我们依次枚举前序遍历中除第一个节点之外的每一个节点，若此时指针index指向栈顶节点，说明此时的index节点是前一个弹出节点的右孩子，我们不断弹出栈顶节点并右移index直到指针index与栈顶节点不同；若此时指针index与栈顶节点不同，则说明当前节点是栈顶节点的左孩子。处理完两种情况之后，我们将当前节点入栈。

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if (!preorder.size()) {
            return nullptr;
        }
        TreeNode* root = new TreeNode(preorder[0]);
        stack<TreeNode*> stk;
        stk.push(root);
        int inorderIndex = 0;
        for (int i = 1; i < preorder.size(); ++i) {
            int preorderVal = preorder[i];
            TreeNode* node = stk.top();
            if (node->val != inorder[inorderIndex]) {
                node->left = new TreeNode(preorderVal);
                stk.push(node->left);
            }
            else {
                while (!stk.empty() && stk.top()->val == inorder[inorderIndex]) {
                    node = stk.top();
                    stk.pop();
                    ++inorderIndex;
                }
                node->right = new TreeNode(preorderVal);
                stk.push(node->right);
            }
        }
        return root;
    }
};