Update similar-string-groups.py

kamyu104 · web-flow · commit 2fe62fa57a31 · 2018-05-22T01:42:06.000+08:00
diff --git a/Python/similar-string-groups.py b/Python/similar-string-groups.py
@@ -82,9 +82,12 @@ def isSimilar(a, b):
                     union_find.union_set(i1, i2)
         else:
             buckets = collections.defaultdict(list)
+            lookup = set()
             for i in xrange(len(A)):
                 word = list(A[i])
-                buckets["".join(word)].append(i)
+                if A[i] not in lookup:
+                    buckets[A[i]].append(i)
+                    lookup.add(A[i])
                 for j1, j2 in itertools.combinations(xrange(L), 2):
                     word[j1], word[j2] = word[j2], word[j1]
                     buckets["".join(word)].append(i)
