zlatan的算法笔记19--从前序与中序遍历序列构造二叉树（力扣105题）

算法笔记

昨天的题目为使用中序和后序遍历构造二叉树，本题与昨天题目类似，是通过前序和中序遍历构造二叉树，如果搞懂了之前的逻辑，本题解起来会轻松许多。

题目描述

解题思路

第一步：如果两个数组大小为零的话，说明是空节点了。

第二步：如果不为空，那么取前序数组第一个元素作为节点元素。

第三步：找到前序数组第一个元素在中序数组的位置，作为切割点

第四步：切割中序数组，切成中序左数组和中序右数组 （顺序别搞反了，一定是先切中序数组）

第五步：切割前序数组，切成前序左数组和前序右数组

第六步：递归处理左区间和右区间

这里与昨天代码有一些不同之处，主要是体现在分割前序数组的地方，我们知道在前序数组当中第一个元素已经被当做了根节点，所以在分割的时候要从第二个元素开始，即preorder.begin() + 1，其他部分就没有什么区别了。

代码部分

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* traversal(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() == 0) {
            return nullptr;
        }

        int rootValue = preorder[0];
        TreeNode* root = new TreeNode(rootValue);

        if(preorder.size() == 1) {
            return root;
        }

        int cutIndex;
        for(cutIndex = 0; cutIndex < inorder.size(); cutIndex++) {
            if(inorder[cutIndex] == rootValue) break;
        }

        vector<int> leftInorder(inorder.begin(), inorder.begin() + cutIndex);
        vector<int> rightInorder(inorder.begin() + cutIndex + 1, inorder.end());

        vector<int> leftPreorder(preorder.begin() + 1, preorder.begin() + 1 + leftInorder.size());
        vector<int> rightPreorder(preorder.begin() + 1 + leftInorder.size(), preorder.end());

        root->left = traversal(leftPreorder, leftInorder);
        root->right = traversal(rightPreorder, rightInorder);

        return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() == 0 || inorder.size() == 0) {
            return nullptr;
        }
        return traversal(preorder, inorder);
    }
};

结语

构造二叉树在思想上相对轻松，但在代码实现部分是有一些难度的，但只要搞清楚实现的思路，多加练习，应该就问题不大。