题目链接:Construct Binary Search Tree from Preorder Traversal
题目大意:给定一个二叉树的前序遍历,要求给出对应的二叉搜索树
题目思路:我们可以知道,对于给定的前序遍历,遍历的顺序是前左右,所以对于一个前序序列,第一个数一定就是根节点,然后后面有一段序列是左子树,剩下的序列是右子树,又因为是二叉搜索树,左子树的所有节点都小于根节点,右子树的所有节点都大于根节点,所以我们只需要找到第一个大于根节点的值,这个值就是我们想要的右子树的根节点,然后对于新的左子树和右子树,可以接着使用同样的方法进行递归,类似于分治的做法进行操作
时间复杂度&&空间复杂度:O(n)&&O(1)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
void dfs(vector<int>& preorder,TreeNode* root,int left,int right){
if(left > right) {
root->left = NULL;
root->right = NULL;
return ;
}
if(preorder[left] > root->val){
root->left = NULL;
root->right = new TreeNode(preorder[left]);
dfs(preorder,root->right,left+1,right);
}
else{
int i;
for(i = left;i <= right;i++){
if(preorder[i] > root->val){
root->left = new TreeNode(preorder[left]);
dfs(preorder,root->left,left+1,i-1);
root->right = new TreeNode(preorder[i]);
dfs(preorder,root->right,i+1,right);
break;
}
}
if(i == right+1){
root->left = new TreeNode(preorder[left]);
dfs(preorder,root->left,left+1,i-1);
}
}
}
TreeNode* bstFromPreorder(vector<int>& preorder) {
int len = preorder.size();
TreeNode* root = new TreeNode(preorder[0]);
dfs(preorder,root,1,len-1);
return root;
}
};