Skip to content

Commit 427323d

Browse files
authored
Update binary_search_tree.py
1 parent 3ab69f2 commit 427323d

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

data_structures/binary_tree/binary_search_tree.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -91,9 +91,9 @@
9191

9292
from __future__ import annotations
9393

94+
from collections.abc import Iterator
9495
from dataclasses import dataclass
9596
from pprint import pformat
96-
from collections.abc import Iterator # fix UP035
9797

9898

9999
@dataclass
@@ -130,7 +130,8 @@ def __bool__(self) -> bool:
130130
def __iter__(self) -> Iterator[int]:
131131
if self.root:
132132
yield from self.root
133-
return iter(())
133+
else:
134+
yield from ()
134135

135136
def __str__(self) -> str:
136137
return str(self.root) if self.root else "Empty tree"
@@ -175,14 +176,13 @@ def insert(self, *values: int) -> BinarySearchTree:
175176
for value in values:
176177
self.__insert(value)
177178
return self
178-
179179
def search(self, value: int) -> Node | None:
180180
if self.empty():
181181
raise IndexError("Warning: Tree is empty! please use another.")
182182

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

0 commit comments

Comments
 (0)