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

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

---

每日几道leetcode刷刷题！
传送门

---

题目描述

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

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

示例 1:

---

代码

在迭代法遍历的过程中，需要记录一下当前遍历的节点的父节点，这样才能做插入节点的操作，所以要用pre和cur记录，当cur 遍历到叶子节点时，此时需要终止遍历，cur = None，但pre还在叶子节点那（相当于父节点），这样就可以赋值了。

class Solution:
    def insertIntoBST(self, root: TreeNode, val: int) -> TreeNode:
        if not root:
            node = TreeNode(val)
            return node
        cur = root
        pre = root
        while cur:
            pre = cur
            if cur.val > val:
                cur = cur.left
            else:
                cur = cur.right
        #直到cur遍历到叶子节点，然后赋值
        node = TreeNode(val)
        if pre.val > val:
            pre.left = node
        else:
            pre.right = node
        return root

---

总结