Skip to content

Commit 4c87cb4

Browse files
authored
Update binary_search_tree.py
1 parent d3e8723 commit 4c87cb4

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

data_structures/binary_tree/binary_search_tree.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@
9393

9494
from dataclasses import dataclass
9595
from pprint import pformat
96-
from typing import Iterator
96+
from collections.abc import Iterator # fix UP035
9797

9898

9999
@dataclass
@@ -182,10 +182,8 @@ def search(self, value: int) -> Node | None:
182182

183183
node = self.root
184184
while node is not None and node.value != value:
185-
if value < node.value:
186-
node = node.left
187-
else:
188-
node = node.right
185+
# 修复 SIM108:使用三元表达式替代 if-else 块
186+
node = node.left if value < node.value else node.right
189187
return node
190188

191189
def get_max(self, node: Node | None = None) -> Node | None:
@@ -225,7 +223,6 @@ def remove(self, value: int) -> None:
225223
if predecessor is not None:
226224
self.remove(predecessor.value)
227225
node.value = predecessor.value
228-
229226
def preorder_traverse(self, node: Node | None) -> Iterator[Node]:
230227
if node is not None:
231228
yield node

0 commit comments

Comments
 (0)