LeetCode——105. Construct Binary Tree from Preorder and Inorder Traversal(C++)
Description:
Given two integer arrays preorder and inorder where preorder is the preorder traversal of a binary tree and inorder is the inorder traversal of the same tree, construct and return the binary tree.
Example 1:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
Example 2:
Input: preorder = [-1], inorder = [-1]
Output: [-1]
Constraints:
1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder and inorder consist of unique values.
Each value of inorder also appears in preorder.
preorder is guaranteed to be the preorder traversal of the tree.
inorder is guaranteed to be the inorder traversal of the tree.
题目解析:
给出中序遍历和先序遍历的数组,返回该二叉树;
代码:
/**
* 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 {
TreeNode* BuildTree(vector<int>& in,vector<int>& pre,int inleft,int inright,int preleft,int preright)
{
if(inleft>inright)return NULL;
TreeNode* cur=new TreeNode;
cur->val=pre[preleft];
int ind=inleft;
while(in[ind]!=pre[preleft])ind++;
cur->left=BuildTree(in,pre,inleft,ind-1,preleft+1,ind-inleft+preleft);
cur->right=BuildTree(in,pre,ind+1,inright,ind-inleft+preleft+1,preright);
return cur;
}
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int sz=preorder.size();
return BuildTree(inorder,preorder,0,sz-1,0,sz-1);
}
};
Submission: