return child:parent dict for UCS and A*

petercorke · petercorke · commit eab98c721e8b · 2022-07-06T17:28:34.000+10:00
diff --git a/pgraph/PGraph.py b/pgraph/PGraph.py
@@ -1113,11 +1113,15 @@ def path_UCS(self, S, G, verbose=False, summary=False):
             path.insert(0, p)
             x = p
 
+        parent_names = {}
+        for v, p in parent.items():
+            parent_names[v.name] = p.name
+
         if summary or verbose:
             print(
                 f"{len(explored)} vertices explored, {len(frontier)} remaining on the frontier")
 
-        return path, length, parent
+        return path, length, parent_names
 
     def path_Astar(self, S, G, verbose=False, summary=False):
         """
@@ -1210,11 +1214,15 @@ def path_Astar(self, S, G, verbose=False, summary=False):
             path.insert(0, p)
             x = p
 
+        parent_names = {}
+        for v, p in parent.items():
+            parent_names[v.name] = p.name
+
         if summary or verbose:
             print(
                 f"{len(explored)} vertices explored, {len(frontier)} remaining on the frontier")
 
-        return path, length, parent
+        return path, length, parent_names
 
 
 # -------------------------------------------------------------------------- #
