代码随想录二刷 ｜ 二叉树 ｜ 二叉搜索树中的插入操作

题目描述

701.二叉搜索树中的插入操作

给定二叉搜索树（BST）的根节点和要插入树中的值，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。

提示：

• 给定的树上的节点数介于 0 和 10^4 之间
• 每个节点都有一个唯一整数值，取值范围从 0 到 10^8
• -10^8 <= val <= 10^8
• 新值和原始二叉搜索树中的任意节点值都不同

解题思路

遍历二叉树上结构，找到空节点并插入即可。

递归三部曲

1. 确定递归函数的参数和返回值
   参数为根节点和要插入的值
   返回值类型为TreeNode*

   TreeNode* insertIntoBST(TreeNode* root, int val)
   

2. 确定递归函数的终止条件
   当找到空节点时，就找到了要插入的位置，终止递归并把插入的节点返回

   if (root == NULL) {
   	TreeNode* node = new TreeNode(val);
   	return ndoe;	
   }
   

3. 确定单层递归的逻辑
   BST只需要遍历边，它可以根据插入元素的数值，决定递归方向。

   if (root->val > val) root->left = insertIntoBST(root->left, val);
   if (root->val < val) root->right = insertIntoBST(root->right, val);
   return root;
   

代码实现

class Solution {
public:
	TreeNode* insertIntoBST(TreeNode* root, int val){
		if (root == NULL) {
			TreeNode* node = new TreeNode(val);
			return node;
		}
		if (root->val > val) root->left = insertIntoBST(root->left, val);
		if (root->val < val) root->right = insertIntoBST(root->right, val);
		return root;
	}
};