leetcode 701 学习笔记（递归、迭代）

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

问题描述：传送门

思路：

本来以为这题挺简单的，但读完题目发现这棵二叉树有多种有效的插入方式，还可以重构二叉搜索树。我直接无语…
…
但换个思路，先不考虑题目中改变树的结构的插入方式。
我们可以先按照二叉搜索树的规则去遍历，遇到空节点再插入节点就可以了。
其算法演示动画如下：

例如插入元素10，只用找到这棵二叉树的末尾节点便可， 不需要调整二叉树的结构。

1、递归

① 确定递归函数的参数和返回值
参数：根节点指针，以及要插入的元素。
返回值：返回新加入节点与其父节点的赋值。
返回类型： TreeNode * 。

TreeNode* insertIntoBST(TreeNode* root, int val) 

② 确定终止条件
找到遍历的节点为null时，为要插入节点，把插入的节点返回。

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;

这里就完成了新加入节点确认父子关系的操作，root->left和root->right都收到了新节点。

整体代码如下：

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;
    }
};

本题也可以不需要返回值，当在二叉树中找到插入的节点位置，直接让父节点指向插入节点，结束递归也可以。

如下：

TreeNode* parent; // 记录遍历节点的父节点
void traversal(TreeNode* cur, int val)

但没有返回值的话，就需要记录上一个节点（parent），遇到空节点了，就让parent左孩子或者右孩子指向新插入的节点。然后结束递归。
如下：

class Solution {
private:
    TreeNode* parent;
    void traversal(TreeNode* cur, int val) {
        if (cur == NULL) {
            TreeNode* node = new TreeNode(val);
            if (val > parent->val) 
            	parent->right = node;
            else 
            	parent->left = node;
            return;
        }
        parent = cur;
        if (cur->val > val) 
        	traversal(cur->left, val);
        if (cur->val < val) 
        	traversal(cur->right, val);
        return;
    }

public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        parent = new TreeNode(0);
        if (root == NULL) {
            root = new TreeNode(val);
        }
        traversal(root, val);
        return root;
    }
};

2、迭代

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (root == NULL) {
            TreeNode* node = new TreeNode(val);
            return node;
        }
        TreeNode* cur = root;
        TreeNode* parent = root;
         // 需要记录上一个节点，否则无法赋值新节点
        while (cur != NULL) {
            parent = cur;
            if (cur->val > val) cur = cur->left;
            else cur = cur->right;
        }
        TreeNode* node = new TreeNode(val);
        if (val < parent->val) 
        	parent->left = node;
        // 此时是用parent节点的进行赋值
        else parent->right = node;
        return root;
    }
};

迭代法就需要记录当前遍历节点的父节点了，这个和没有返回值的递归函数实现的代码逻辑是一样的。