Skip to content

Commit 0a897c3

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 9fc373d commit 0a897c3

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

data_structures/heap/fibonacci_heap.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@ class FibonacciTree:
1515

1616
def __init__(self, key: int) -> None:
1717
self.key = key
18-
self.children: list['FibonacciTree'] = []
18+
self.children: list["FibonacciTree"] = []
1919
self.order = 0
2020

21-
def add_at_end(self, child_node: 'FibonacciTree') -> None:
21+
def add_at_end(self, child_node: "FibonacciTree") -> None:
2222
"""
2323
Adds a child node 'child_node' to the end of the children list.
2424
@@ -53,8 +53,8 @@ class FibonacciHeap:
5353
"""
5454

5555
def __init__(self) -> None:
56-
self.trees: list['FibonacciTree'] = []
57-
self.least: 'FibonacciTree' | None = None
56+
self.trees: list["FibonacciTree"] = []
57+
self.least: "FibonacciTree" | None = None
5858
self.count = 0
5959

6060
def insert(self, key: int) -> None:
@@ -66,7 +66,7 @@ def insert(self, key: int) -> None:
6666
"""
6767
new_tree = FibonacciTree(key)
6868
self.trees.append(new_tree)
69-
if (self.least is None or key < self.least.key):
69+
if self.least is None or key < self.least.key:
7070
self.least = new_tree
7171
self.count = self.count + 1
7272

@@ -88,8 +88,7 @@ def extract_min(self) -> int | None:
8888
Returns:
8989
int: The minimum key.
9090
"""
91-
smallest = self.least
92-
if smallest is not None:
91+
if (smallest := self.least) is not None:
9392
for child in smallest.children:
9493
if child is not None:
9594
self.trees.append(child)
@@ -126,9 +125,10 @@ def consolidate(self) -> None:
126125
for k in aux:
127126
if k is not None:
128127
self.trees.append(k)
129-
if (self.least is None or k.key < self.least.key):
128+
if self.least is None or k.key < self.least.key:
130129
self.least = k
131130

131+
132132
def floor_log2(x: int) -> int:
133133
"""
134134
Computes the floor of the base-2 logarithm of 'x'.
@@ -141,11 +141,14 @@ def floor_log2(x: int) -> int:
141141
"""
142142
return math.floor(math.log2(x)) if x > 0 else 0
143143

144+
144145
# Doctest for floor_log2
145146
if __name__ == "__main__":
146147
import doctest
148+
147149
doctest.testmod()
148150

149151
if __name__ == "__main__":
150152
import doctest
153+
151154
doctest.testmod()

0 commit comments

Comments
 (0)