【剑指Offer记录】7-重建二叉树

Part1. 我的思路和代码

前序遍历是“根-左-右”，中序遍历是“左-根-右”。从前序遍历preOrder的第一个值可以得到根，然后以该值为分界，可以把中序遍历分成2个部分，记分界的下标是idx，那么左边的部分[0, idx)（左闭右开区间，后同）是左子树的中序遍历，右边的部分[idx+1, vinOrder.size)是右子树的中序遍历。由于同一棵树的前序遍历和中序遍历，包含的节点数是一样的，因此前序遍历的[1, 1+(idx-0))部分为左子树的前序遍历，前序遍历的[1+(idx-0), preOrder.size()]部分为右子树的前序遍历。用图表示如下（做题笔记中的第一张自制图）：

根据以上思路，采用递归方式，递归思想为：先建立根，然后分别建立左子树、右子树，建立左、右子树时函数参数传入计算得到的子树的前序、中序遍历数组。

递归调用的总次数为节点数n，时间复杂度为O(n)，对于每个节点的处理，都需要找到中序遍历数组中分界的索引，时间复杂度也为O(n)，因此总的时间复杂度我认为是O(n^2)，题目要求为O(n)，但提交后可以通过。

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
TreeNode* reConstructBinaryTree(vector<int>& preOrder, vector<int>& vinOrder) {
    // write code here
    TreeNode *root;
    vector<int> leftPre;
    vector<int> leftIn;
    vector<int> rightPre;
    vector<int> rightIn;
    int idx;

    if(preOrder.empty()){
        root = nullptr;
    }
    else{
        root = new TreeNode(preOrder[0]); // 建立根节点
        for(idx=0; idx<vinOrder.size(); idx++){
            if(vinOrder[idx] == preOrder[0]){
                break;
            }
        }
        for(int i=1; i<idx+1; i++){ // 计算左子树的前序和中序遍历
            leftPre.push_back(preOrder[i]);
        }
        for(int i=0; i<idx; i++){
            leftIn.push_back(vinOrder[i]);
        }
        for(int i=idx+1; i<preOrder.size(); i++){ // 计算右子树的前序和中序遍历
            rightPre.push_back(preOrder[i]);
        }
        for(int i=idx+1; i<vinOrder.size(); i++){
            rightIn.push_back(vinOrder[i]);
        }
        if(leftIn.empty()){ // 建立左子树
            root->left = nullptr;
        }
        else{
            root->left = reConstructBinaryTree(leftPre, leftIn);
        }
        if(rightIn.empty()){ // 建立右子树
            root->right = nullptr;
        }
        else{
            root->right = reConstructBinaryTree(rightPre, rightIn);
        }
    }

    return root;
}

特别记录：我认为以上做法时间复杂度为O(n^2)，不是题目要求的O(n)，有待思考。

Part2. 心得体会

• 知识储备对做题还是影响很大的，印象里大二数据结构课程学习了如何根据前序遍历和中序遍历构造二叉树，但今天做这道题目时已经记忆模糊，想了半个小时才逐渐想起。
• 递归思路为：大问题分解为可以用相同做法解决的小问题，如在该题目中，建立左、右子树的过程，同样是根据前序、中序遍历数组。