Tidy up Day 12.

pak21 · Dec 12, 2017 · b93680a · b93680a
1 parent 7ab510e
commit b93680a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/12/12.py b/12/12.py
@@ -21,16 +21,16 @@
     groups += 1
     first = all_items.pop()
     todo = collections.deque([first])
-    seen = set()
+    seen = set([first])
 
     while len(todo) > 0:
         item = todo.popleft()
-        if item in all_items:
-            all_items.remove(item)
-        seen.add(item)
         for r in connections[item]:
             if not r in seen:
                 todo.append(r)
+                seen.add(r)
+
+    all_items -= seen
 
     print('{} -> {}'.format(first, len(seen)))