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

题目描述

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

问题分析

递归法： 在二叉搜索树中插入节点，通过与节点值比较判断，如果小于节点值，就往左子树方向递归，大于节点值，就往右子树方向递归，直到节点为空，就找到插入位置了，建立新节点插入该位置。题目要求要返回根节点，可以把添加节点返回给上一层，，递归返回时都向上一层返回节点。
迭代法： 方法逻辑与递归法类似，只是要注意用pre来记录上一个节点，这样才能在找插入位置时，插入新节点。

代码实现

// 编程软件：VS2019
// 参考书籍：代码随想录
#include<iostream>
using namespace std;

// 定义二叉树
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int v):val(v),left(nullptr),right(nullptr){}
};

/*
* 递归法：在二叉搜索树中插入节点，通过与节点值比较判断，
* 如果小于节点值，就往左子树方向递归，大于节点值，就往右子树方向递归，
* 直到节点为空，就找到插入位置了，建立新节点插入该位置。
* 题目要求要返回根节点，可以把添加节点返回给上一层，递归返回都向上一层返回节点。
*/ 
TreeNode* insertIntoBST1(TreeNode* root, int val) {
	if (root == NULL) {
		TreeNode* node = new TreeNode(val); // 建立新节点
		return node; // 返回插入节点
	}
	if (val < root->val) root->left = insertIntoBST1(root->left, val);// 小于，往左子树方向
	if (val > root->val) root->right = insertIntoBST1(root->right, val); // 大于，往右子树方向
	return root; // 向上一层返回节点
}

// 迭代法：方法逻辑与递归法类似，只是要注意用pre来记录上一个节点，这样才能在找插入位置时，插入新节点。
TreeNode* insertIntoBST2(TreeNode* root, int val) {
	if (root == NULL) { // 如果是一颗空树，直接建立节点为根节点
		TreeNode* node = new TreeNode(val);
		return node;
	}
	TreeNode* cur = root;
	// 这里很重要，需要记录上一个节点，否则无法插入新节点
	TreeNode* pre = root;
	// 与递归逻辑一样的
	while (cur != NULL) {
		pre = cur;
		if (val < cur->val) cur = cur->left;
		else cur = cur->right;
	}
	TreeNode* node = new TreeNode(val); // 建立新节点
	// 使用pre插入新节点节点
	if (val < pre->val) pre->left = node; 
	else pre->right = node;
	return root;
}


int main() {
	TreeNode* root = new TreeNode(7);
	root->left = new TreeNode(4);
	root->right = new TreeNode(8);
	root->left->left = new TreeNode(3);
	root->left->right = new TreeNode(5);
	TreeNode* res = insertIntoBST2(root,6);
	cout << res->val << endl;
}
// 结果：7