题目来源: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;
}
};