Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
similar-string-groups.py 1.93 KB
一键复制 编辑 原始数据 按行查看 历史
Allen Liu 提交于 2018-10-13 13:15 +08:00 . remove unused code
# Time: O(n^2 * l) ~ O(n * l^4)
# Space: O(n) ~ O(n * l^3)
import collections
import itertools
class UnionFind(object):
def __init__(self, n):
self.set = range(n)
self.__size = n
def find_set(self, x):
if self.set[x] != x:
self.set[x] = self.find_set(self.set[x]) # path compression.
return self.set[x]
def union_set(self, x, y):
x_root, y_root = map(self.find_set, (x, y))
if x_root == y_root:
return False
self.set[min(x_root, y_root)] = max(x_root, y_root)
self.__size -= 1
return True
def size(self):
return self.__size
class Solution(object):
def numSimilarGroups(self, A):
def isSimilar(a, b):
diff = 0
for x, y in itertools.izip(a, b):
if x != y:
diff += 1
if diff > 2:
return False
return diff == 2
N, L = len(A), len(A[0])
union_find = UnionFind(N)
if N < L*L:
for (i1, word1), (i2, word2) in \
itertools.combinations(enumerate(A), 2):
if isSimilar(word1, word2):
union_find.union_set(i1, i2)
else:
buckets = collections.defaultdict(list)
lookup = set()
for i in xrange(len(A)):
word = list(A[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)
word[j1], word[j2] = word[j2], word[j1]
for word in A: # Time: O(n * l^4)
for i1, i2 in itertools.combinations(buckets[word], 2):
union_find.union_set(i1, i2)
return union_find.size()
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助