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)