701. 二叉搜索树中的插入操作
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
class Solution {
public:
//迭代法
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(root == nullptr) return new TreeNode(val);
TreeNode *cur = root;
while(cur){
if(cur->val > val) {
pre = cur;
cur = cur->left;
if(cur == nullptr)
{
cur = new TreeNode(val);
pre->left = cur;
return root;
}
continue;
}
if(cur->val < val) {
pre = cur;
cur = cur->right;
if(cur == nullptr)
{
cur = new TreeNode(val);
pre->right = cur;
return root;
}
continue;
}
}
return root;
}
//递归
TreeNode* insertIntoBST(TreeNode* root, int val){
if(root == nullptr) return new TreeNode(val);
if(val < root->val){
TreeNode *left = insertIntoBST(root->left,val);
root->left = left;
}
if(val > root->val){
TreeNode *right = insertIntoBST(root->right,val);
root->right = right;
}
return root;
}
private:
TreeNode *pre = nullptr;
};
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理