题目要求
给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
示例 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