Merge pull request #2714 from wq-win/master

修改 0701.二叉搜索树中的插入操作,增加python版本的迭代法(精简),修正 0701.二叉搜索树中的插入操作,递归法版本二应该是迭代法
This commit is contained in:
程序员Carl
2024-09-07 16:08:47 +08:00
committed by GitHub

View File

@ -283,32 +283,10 @@ class Solution:
return TreeNode(val)
self.traversal(root, val)
return root
```
递归法(版本二)
```python
class Solution:
def insertIntoBST(self, root, val):
if root is None:
return TreeNode(val)
parent = None
cur = root
while cur:
parent = cur
if val < cur.val:
cur = cur.left
else:
cur = cur.right
if val < parent.val:
parent.left = TreeNode(val)
else:
parent.right = TreeNode(val)
return root
```
递归法(版本三)
```python
class Solution:
def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
if root is None or root.val == val:
@ -326,7 +304,7 @@ class Solution:
return root
```
递归法(版本
递归法(版本
```python
class Solution:
def insertIntoBST(self, root, val):
@ -340,10 +318,9 @@ class Solution:
root.right = self.insertIntoBST(root.right, val)
return root
```
迭代法
迭代法(版本一)
```python
class Solution:
def insertIntoBST(self, root, val):
@ -367,9 +344,52 @@ class Solution:
parent.right = node # 将新节点连接到父节点的右子树
return root
```
迭代法(版本二)
```python
class Solution:
def insertIntoBST(self, root, val):
if root is None:
return TreeNode(val)
parent = None
cur = root
while cur:
parent = cur
if val < cur.val:
cur = cur.left
else:
cur = cur.right
if val < parent.val:
parent.left = TreeNode(val)
else:
parent.right = TreeNode(val)
return root
```
迭代法(精简)
```python
class Solution:
def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
if not root: # 如果根节点为空,创建新节点作为根节点并返回
return TreeNode(val)
cur = root
while cur:
if val < cur.val:
if not cur.left: # 如果此时父节点的左子树为空
cur.left = TreeNode(val) # 将新节点连接到父节点的左子树
return root
else:
cur = cur.left
elif val > cur.val:
if not cur.right: # 如果此时父节点的左子树为空
cur.right = TreeNode(val) # 将新节点连接到父节点的右子树
return root
else:
cur = cur.right
```
-----
### Go