Skip to content

Commit 16bb0d5

Browse files
committed
fix the reviewer comment
1 parent e2cf889 commit 16bb0d5

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

Sprint-2/implement_skip_list/skip_list.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,10 @@ def _find_position(self, value):
4343
b = self.skips[i + 1]
4444

4545
node_a = self._get_node(a)
46-
node_b = self._get_node(b)
46+
node_b = node_a
47+
for _ in range(b - a):
48+
if node_b:
49+
node_b = node_b.next
4750

4851
if node_a.value <= value < node_b.value:
4952
# linear search between a and b
@@ -95,7 +98,10 @@ def __contains__(self, value):
9598
b = self.skips[i + 1]
9699

97100
node_a = self._get_node(a)
98-
node_b = self._get_node(b)
101+
node_b = node_a
102+
for _ in range(b - a):
103+
if node_b:
104+
node_b = node_b.next
99105

100106
if node_a.value <= value < node_b.value:
101107
idx = a

0 commit comments

Comments
 (0)