diff --git a/data_structures/linked_list/has_loop.py b/data_structures/linked_list/has_loop.py index a25204e3c4ce..70c9aed96fbb 100644 --- a/data_structures/linked_list/has_loop.py +++ b/data_structures/linked_list/has_loop.py @@ -37,13 +37,17 @@ def has_loop(self) -> bool: >>> root_node.has_loop True """ - try: - list(self) - return False - except ContainsLoopError: - return True + slow = self + fast = self + while fast is not None and fast.next_node is not None: + slow = slow.next_node + fast = fast.next_node.next_node + if slow is fast: + return True + + return False if __name__ == "__main__": root_node = Node(1) root_node.next_node = Node(2)