txlyre 3 éve
szülő
commit
4e29d2eed4
1 módosított fájl, 3 hozzáadás és 0 törlés
  1. 3 0
      yafn/__main__.py

+ 3 - 0
yafn/__main__.py

@@ -150,6 +150,9 @@ def distance(graph, start, end):
   visited = []
   queue = [[start]]
 
+  if start == end:
+    return 0
+
   while queue:
     path = queue.pop(0)
     node = path[-1]