Leetcode 893. 特殊等價字串組 python 版本

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)