题目来源:https://leetcode.cn/problems/construct-binary-tree-from-preorder-and-inorder-traversal/description/
C++题解:前序遍历是中左右,中序遍历是左中右,所以拿到两个遍历数组,我们可以从前序遍历首节点获取中间节点,再通过中间节点在中序遍历的索引,可以将中序遍历的左右子树分割开,根据左子树的长度也可以将前序遍历的左右子树分割开。
进行遍历时,我们需要分别找到左右子树,所以每次遍历,都需要单独获取左子树的前序和中序遍历数组、右子树的前序和中序遍历数组。当无左右子树时,返回空指针;当左右子树为叶子节点时,返回当前节点。
/*** 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* buildTree(vector<int>& preorder, vector<int>& inorder) {int len = preorder.size();if(len == 0) return nullptr;TreeNode* root = new TreeNode(preorder[0]);if(len == 1) return root;int ind = find(inorder.begin(), inorder.end(), preorder[0]) - inorder.begin();vector<int> lepre(preorder.begin() + 1, preorder.begin() + ind + 1);vector<int> lein(inorder.begin(), inorder.begin() + ind);root->left = buildTree(lepre, lein);vector<int> ripre(preorder.begin() + ind + 1, preorder.end());vector<int> riin(inorder.begin() + ind + 1, inorder.end());root->right = buildTree(ripre, riin);return root;}
};