力扣 105. 从前序与中序遍历序列构造二叉树

题目

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

和这个题目类似的还有 106. 从中序与后序遍历序列构造二叉树

解题

看到题目以后，就想到上数据结构的时候利用树的先（后）序遍历和中序遍历来搭建树的方法，但是不知道如何用算法实现，惭愧！

递归

1. 似乎树这个数据结构和递归就很有缘分，从构建一棵树到遍历一棵树，都可以用遍历来解决，以后碰见树的题目可以先想想能不能用递归解决！
2. 我们来分析一下用先序遍历序列和中序遍历序列怎样构建一棵树。
   

• 先（前）序遍历序列的第一个肯定是根节点；
• 在中序遍历序列中找处根节点，则可以判断，根节点左边的是左子树，右边的是右子树。
• 根据找到的左子树和右子树的长度，也就可以确定先（前）序遍历序列中左子树的先序遍历序列和右子树的先序遍历序列
• 然后回到第一步，直到没有左子树和右子树的序列，这样整棵树就构建出来了。

class Solution {
private:
    unordered_map<int, int> index;//利用map表来记录每个节点在中序遍历序列中所在的位置，简化查询，简化后查询的时间复杂度就为O(1)
public:
    /**
    *pre_left是先序遍历序列中的左端点
    *pre_right是先序遍历序列中的右端点
    *in_left是中序遍历序列中的左端点
    *in_right是中序遍历序列中的右端点
    */
    TreeNode* find(vector<int> preorder, vector<int> inorder, int pre_left, int pre_right, int in_left, int in_right){
        if(pre_left > pre_right) return NULL;

        //在先序遍历中找到根节点
        int root = preorder[pre_left];
        //找到根节点在中序遍历序列中的位置，以此来确定左子树和右子树
        int in_root = index[root];
        //左子树长度
        int leftlength = in_root - in_left;

        //建立根节点
        TreeNode* tree = new TreeNode(root);
        //建立左子树
        tree->left = find(preorder,inorder,pre_left + 1,pre_left + leftlength,in_left ,in_root - 1);
        //建立右子树
        tree->right = find(preorder,inorder,pre_left + leftlength + 1,pre_right,in_root + 1,in_right);

        return tree;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n = inorder.size();
        for(int i = 0; i < n; i++){
             index[inorder[i]] = i;
        }

        return find(preorder,inorder,0,n-1,0,n-1);
    }
};

时间复杂度为O(n)，空间复杂度为O(n)。

至于为什么用unordered_map而不是用map，主要是因为两者背后的原理不同，前者更方便查询。参考map与unordered_map的区别。