Leetcode# 701. 二叉搜索树中的插入操作
701. 二叉搜索树中的插入操作
给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
示例 1:
输入:root = [4,2,7,1,3], val = 5 输出:[4,2,7,1,3,5] 解释:另一个满足题目要求可以通过的树是:
示例 2:
输入:root = [40,20,60,10,30,50,70], val = 25 输出:[40,20,60,10,30,50,70,null,null,25] 示例 3:
输入:root = [4,2,7,1,3,null,null,null,null,null,null], val = 5 输出:[4,2,7,1,3,5]
提示:
给定的树上的节点数介于 0 和 10^4 之间 每个节点都有一个唯一整数值,取值范围从 0 到 10^8 -10^8 <= val <= 10^8 新值和原始二叉搜索树中的任意节点值都不同
本题第一想法用递归解决,思路为判断结点是否为NULL,为空则在此处插入待插入结点。不空则与待插入val值比较,若大于当前节点中的值,则向当前节点right递归调用自身函数,小于同理。(赋第一次代码(有问题))
class Solution { public: void insert(TreeNode* root, int val){ if(!root){ root = new TreeNode(val); return; } else if(root -> val > val) insert(root -> left, val); else if(root -> val < val) insert(root -> right, val); } TreeNode* insertIntoBST(TreeNode* root, int val) { insert(root, val); return root; } };
开始时遇到问题,递归完成后新建结点没有插入树中,测试时输出中无插入的节点。 后发现问题在于新建的结点没有与树连接,从树中无法访问到插入的节点。 之后对其进行修改,删去void函数,直接在TreeNode* insertIntoBST函数中实现操作。因为这样有返回值,可以通过将新建的结点返回并赋值给他的父节点得左子树或右子树。(赋修改后代码(无问题))
class Solution { public: TreeNode* insertIntoBST(TreeNode* root, int val) { if(!root){ return new TreeNode(val); } else if(root -> val > val) root -> left = insertIntoBST(root -> left, val); else if(root -> val < val) root -> right = insertIntoBST(root -> right, val); return root; } };