面试题 10.02. 变位词组
题目描述:
编写一种方法,对字符串数组进行排序,将所有变位词组合在一起。变位词是指字母相同,但排列不同的字符串。
注意:本题相对原题稍作修改
示例:
输入: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”],
输出:
[
[“ate”,“eat”,“tea”],
[“nat”,“tan”],
[“bat”]
]
提示:
- 所有输入均为小写字母。
- 不考虑答案输出的顺序。
解法
本题的关键在于如何为字符串创建唯一标识,以作为字典的key,最直接的思路是对字符串排序,排序后的字符串就是唯一标识;或者统计各个字母出现的次数,然后把字符及其次数连接起来作为唯一标识。剩余的比较简单不再赘述。
代码
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
str2strs = {}
for sstr in strs:
sort_str = str(sorted(sstr))
if sort_str in str2strs:
str2strs[sort_str].append(sstr)
else:
str2strs[sort_str] = [sstr]
return list(str2strs.values())
测试结果
执行用时:52 ms, 在所有 Python3 提交中击败了 51.58% 的用户
内存消耗:18 MB, 在所有 Python3 提交中击败了 36.35% 的用户
说明
算法题来源:力扣(LeetCode)