improve 01_breadth-first_search.py (egonSchiele#195)

dayyass · web-flow · commit c1c68d9d2a54 · 2021-03-14T10:39:59.000-05:00
diff --git a/06_breadth-first_search/python/01_breadth-first_search.py b/06_breadth-first_search/python/01_breadth-first_search.py
@@ -16,8 +16,8 @@ def person_is_seller(name):
 def search(name):
     search_queue = deque()
     search_queue += graph[name]
-    # This array is how you keep track of which people you've searched before.
-    searched = []
+    # This is how you keep track of which people you've searched before.
+    searched = set()
     while search_queue:
         person = search_queue.popleft()
         # Only search this person if you haven't already searched them.
@@ -28,7 +28,7 @@ def search(name):
             else:
                 search_queue += graph[person]
                 # Marks this person as searched
-                searched.append(person)
+                searched.add(person)
     return False
 
 search("you")
