目录
问题描述
给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
示例 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] 仅包含小写字母
计数法:
通过计算每个字母的个数来判断两个单词是不是异位词。
因为我没有想到用哈希表来存储值,最后提交后超时没能通过。
所以这里提醒:一旦需要根据特征进行归类,就应该利用散列表。
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
n = len(strs)
if n == 1:
return [strs]
def same(str1, str2):
if str1 == str2:
return True
count1 = {}
count2 = {}
for i in str1:
count1[i] = count1.get(i,0) + 1
for i in str2:
count2[i] = count2.get(i,0) + 1
if len(count1) == 0 or len(count2)==0 or len(count1) != len(count2):
return False
for k, v in count1.items():
if count2.get(k,0) != v:
return False
return True
result = []
while(len(strs) > 0):
tmp = []
n = len(strs)
str1 = strs[0]
tmp.append(str1)
if n > 1:
for i in range(1,n):
if same(str1, strs[i]):
tmp.append(strs[i])
result.append(tmp)
for t in tmp:
strs.remove(t)
return result
计数法(用哈希表):
由于互为字母异位词的两个字符串包含的字母相同,因此两个字符串中的相同字母出现的次数一定是相同的,故可以将每个字母出现的次数使用字符串表示,作为哈希表的键。
由于字符串只包含小写字母,因此对于每个字符串,可以使用长度为 26 的数组记录每个字母出现的次数
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
mp = collections.defaultdict(list)
for st in strs:
counts = [0] * 26
for ch in st:
counts[ord(ch) - ord("a")] += 1
# 需要将 list 转换成 tuple 才能进行哈希
mp[tuple(counts)].append(st)
return list(mp.values())
时间复杂度: O ( N k ) O(Nk) O(Nk)
排序法:
把一个字符串拆成 char list 然后按字母进行排序。字母异位词 之间排序后的值是相等的。
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
result = collections.defaultdict(list)
for str in strs:
chars = ''.join(sorted(str))
result[chars].append(str)
return list(result.values())
时间复杂度: O ( N k log k ) O(Nk\log{k}) O(Nklogk)
参考文献:
[1] 力扣官方题解