非最优解,仅学习记录。
700.二叉搜索树的搜索
题目要求:
给定二叉搜索树(BST)的根节点 root 和一个整数值 val。
你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 null 。
算法思路:
递归最好
递归遍历所有子树,边遍历边判断是否符合要求值,若符合直接返回结束递归。
!二叉搜索树即二叉排序树,拥有所有左子树均小于根节点,所有右子树均大于根节点的性质
class Solution {
public:
TreeNode* searchBST(TreeNode* root, int val) {
if(root==nullptr)return nullptr;
while(root){
//递归结束出口
if(root->val==val)return root;
//若值小于根节点则去左子树继续找
if(root->val > val) return searchBST(root->left,val);
//若值大于根节点则右子树继续找
if(root->val < val) return searchBST(root->right,val);
}
//所有子树遍历完成均没有符合值则返回空值
return nullptr;
}
};
701.二叉搜索树的插入
题目要求:
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
算法思路:
!二叉搜索树即二叉排序树,左小右大
!注意插入结点意味着需要创建新结点
插入后也要保持左小右大
val跟root比较,若大于则去右子树继续比较,小于则去左子树继续比较
直到遇到左右子树有空位且符合条件时即插入
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
//若树空则直接插入,即创建一个值为val的新结点
if(root==nullptr){
TreeNode* r=new TreeNode(val);
return r;
}
//因为递归,所以当递归到左子树或右子树有空的时候下一次递归就会走上面的判空
if(root->val > val) root->left=insertIntoBST(root->left,val);
if(root->val < val) root->right=insertIntoBST(root->right,val);
return root;
}
};