49.字母异位词分组

136 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路

题目描述

给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"] 输出: [ ["ate","eat","tea"], ["nat","tan"], ["bat"] ] 说明:

所有输入均为小写字母。 不考虑答案输出的顺序。

题目解析

这个题我们依旧拿集合来分析,每个单词都是个集合,那么我们的目的就是找出集合里元素数目和种类都一样的集合。 字母在asscii码中都有对应的数字,因此我们把问题转化为集合里数字的问题,我们可以对每个集合的数字进行排序,也就是对字母进行个排序,排序后,每个单词都会有对应的新的字符列表。

strslist = [(one, tuple(sorted(one))) for one in strs]
#output
[('eat', ('a', 'e', 't')), 
('tea', ('a', 'e', 't')), 
('tan', ('a', 'n', 't')),
 ('ate', ('a', 'e', 't')), 
 ('nat', ('a', 'n', 't')), 
 ('bat', ('a', 'b', 't'))]

我们这时候只需要遍历原列表,把排序后相同的字符列表对应的原字符串添加到同一个列表就可以了,这里我们可以运用python字典的setdefault()方法,以排序后的字符串元组作为key,原字符串作为value,先对原字符串进行分类

 strdict.setdefault(one[1], []).append(one[0])
 #output
 {('a', 'e', 't'): ['eat', 'tea', 'ate'], 
 ('a', 'n', 't'): ['tan', 'nat'], 
 ('a', 'b', 't'): ['bat']}

最后对输出做个格式处理就可以了,字典的值分别添加到列表里面

    for k in strdict.keys():
        strlist.append(strdict[k])

完整代码

def groupAnagrams(strs)
    strdict = {}
    strlist = []
    strslist = [(one, tuple(sorted(one))) for one in strs]
    #print(strslist)
    for one in strslist:
        strdict.setdefault(one[1], []).append(one[0])
    #print(strdict)
    for k in strdict.keys():
        strlist.append(strdict[k])
    return strlist