leetcode刷题日记-字母异位词分组

题目要求

给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

字母异位词 是由重新排列源单词的所有字母得到的一个新单词。

示例 1:

输入: strs = [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出: [[“bat”],[“nat”,“tan”],[“ate”,“eat”,“tea”]]
示例 2:

输入: strs = [“”]
输出: [[“”]]
示例 3:

输入: strs = [“a”]
输出: [[“a”]]

提示：

1 <= strs.length <= 104
0 <= strs[i].length <= 100
strs[i] 仅包含小写字母

解题思路

自己第一遍写的解法比较暴力，超时了之后开始想歪招。我意识到字符串要占用比较大的空间，而且相关操作也很不便。萌生了自创类似ASCII码的想法，但要实现题目要求则必须进行操作后单词特征非常有特点。于是我想到了质数乘积这个思路，只要单词包含的字母一样，通过给字母映射不同的质数，就会导致单词的乘积各不相同，字母异位词又是相同的乘积这一目的

代码实现

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        map = {'a':2,'b':3,'c':5,'d':7,
               'e':11,'f':13,'g':17,'h':19,
               'i':23,'j':29,'k':31,'l':37,
               'm':41,'n':43,'o':47,'p':53,
               'q':59,'r':61,'s':67,'t':71,
               'u':73,'v':79,'w':83,'x':89,'y':97,'z':101,}
        reqMap = {}
        reqList = []
        for str in strs:
            m = 1
            for i in range(len(str)):
                m*=map[str[i]]
            if not m in reqMap:
                reqMap[m] = []
            reqMap[m].append(str)
        return [j for j in reqMap.values()]

相关知识

最后的返回部分用了迭代器x for y in z，意思是对z中的每个元素进行x操作（x应为y的对应形式，如y*2之类）返回值是list