mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-05-12 20:35:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed file name and removed a function call

Browse Source
This commit is contained in:
dbring 2023-10-04 19:03:57 -07:00
parent 4c8450f894
commit 0d1134dac8
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
graphs/deep-clone-graph.py → graphs/deep_clone_graph.py
View File

@ -17,7 +17,7 @@ class Node:
self.neighbors = neighbors if neighbors is not None else []
def clone_graph(node: Node | None) -> None | None:
def clone_graph(node: Node | None) -> Node | None:
"""
This function returns a clone of a connected undirected graph.
>>> clone_graph(Node(1))
@ -33,6 +33,10 @@ def clone_graph(node: Node | None) -> None | None:
originals_to_clones = {} # map nodes to clones
def create_clones(node: Node) -> None:
"""
This helper function populates the originals_to_clones map with
the original nodes in the graph mapped to newly created clones.
"""
if node in originals_to_clones:
return
@ -43,14 +47,11 @@ def clone_graph(node: Node | None) -> None | None:
create_clones(node)
def connect_clones_to_cloned_neighbors() -> None:
for original, clone in originals_to_clones.items():
for neighbor in original.neighbors:
cloned_neighbor = originals_to_clones[neighbor]
clone.neighbors.append(cloned_neighbor)
connect_clones_to_cloned_neighbors()
return originals_to_clones[node]