https://leetcode-cn.com/contest/weekly-contest-99/problems/groups-of-special-equivalent-strings/
題目看起來略微費解,看明白題目後,方法比較簡單,分爲2步:
第一步,找所有字符串,所含字母一樣的字符串。
第二部,找其中奇數位一樣的字符串(或者偶數位)
符合上面兩個條件的是一組。
你將得到一個字符串數組 A
。
如果經過任意次數的移動,S == T,那麼兩個字符串 S
和 T
是特殊等價的。
一次移動包括選擇兩個索引 i
和 j
,且 i%2 == j%2
,並且交換 S[j]
和 S [i]
。
現在規定,A
中的特殊等效字符串組是 A
的非空子集,這樣任何不在 A
中的字符串都不與 A
中的任何字符串特殊等效。
返回 A
中特殊等效字符串組的數量。
示例 1:
輸入:["a","b","c","a","c","c"]
輸出:3
解釋:3 組 ["a","a"],["b"],["c","c","c"]
示例 2:
輸入:["aa","bb","ab","ba"]
輸出:4
解釋:4 組 ["aa"],["bb"],["ab"],["ba"]
示例 3:
輸入:["abc","acb","bac","bca","cab","cba"]
輸出:3
解釋:3 組 ["abc","cba"],["acb","bca"],["bac","cab"]
示例 4:
輸入:["abcd","cdab","adcb","cbad"]
輸出:1
解釋:1 組 ["abcd","cdab","adcb","cbad"]
提示:
1 <= A.length <= 1000
1 <= A[i].length <= 20
- 所有
A[i]
都具有相同的長度。 - 所有
A[i]
都只由小寫字母組成。
class Solution:
def numSpecialEquivGroups(self, A):
"""
:type A: List[str]
:rtype: int
"""
import collections
dict_A = collections.defaultdict(list)
for i in A:
i_array = [j for j in i]
i_ord_array = [n for m, n in enumerate(i) if m % 2 == 0]
i_ord_array.sort()
i_ord_sort_string = ''.join(i_ord_array)
i_array.sort()
i_sort_string = ''.join(i_array)
dict_A[i_sort_string+'_'+i_ord_sort_string].append(i)
return len(dict_A)