forked from shuboc/LeetCode-2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
groups-of-special-equivalent-strings.py
57 lines (54 loc) · 1.48 KB
/
groups-of-special-equivalent-strings.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
# Time: O(n * l)
# Space: O(n)
# You are given an array A of strings.
#
# Two strings S and T are special-equivalent
# if after any number of moves, S == T.
#
# A move consists of choosing two indices i and j with i % 2 == j % 2,
# and swapping S[i] with S[j].
#
# Now, a group of special-equivalent strings from
# A is a non-empty subset S of A such that any string not in S
# is not special-equivalent with any string in S.
#
# Return the number of groups of special-equivalent strings from A.
#
# Example 1:
#
# Input: ["a","b","c","a","c","c"]
# Output: 3
# Explanation: 3 groups ["a","a"], ["b"], ["c","c","c"]
# Example 2:
#
# Input: ["aa","bb","ab","ba"]
# Output: 4
# Explanation: 4 groups ["aa"], ["bb"], ["ab"], ["ba"]
# Example 3:
#
# Input: ["abc","acb","bac","bca","cab","cba"]
# Output: 3
# Explanation: 3 groups ["abc","cba"], ["acb","bca"], ["bac","cab"]
# Example 4:
#
# Input: ["abcd","cdab","adcb","cbad"]
# Output: 1
# Explanation: 1 group ["abcd","cdab","adcb","cbad"]
#
# Note:
# - 1 <= A.length <= 1000
# - 1 <= A[i].length <= 20
# - All A[i] have the same length.
# - All A[i] consist of only lowercase letters.
class Solution(object):
def numSpecialEquivGroups(self, A):
"""
:type A: List[str]
:rtype: int
"""
def count(word):
result = [0]*52
for i, letter in enumerate(word):
result[ord(letter)-ord('a') + 26*(i%2)] += 1
return tuple(result)
return len({count(word) for word in A})