Added tests and docstrings to fibonacci_heap.py

This commit is contained in:
mcawezome 2024-11-18 12:12:17 +08:00
parent 071ce716e4
commit bd943b086a

View File

@ -12,7 +12,6 @@ Operations supported:
- Merge: O(1)
"""
class Node:
"""
A node in a Fibonacci heap.
@ -29,7 +28,6 @@ class Node:
degree: Number of children.
mark: Boolean indicating if node has lost a child.
"""
def __init__(self, val):
self.val = val
self.parent = None
@ -326,7 +324,8 @@ class FibonacciHeap:
node: Starting node for cascading cut.
"""
if parent := node.parent:
parent = node.parent
if parent:
if not node.mark:
node.mark = True
else:
@ -360,5 +359,4 @@ class FibonacciHeap:
if __name__ == "__main__":
import doctest
doctest.testmod()