Leetcode - 49. 字母异位词分组

转载

mb5fe18e7c44408 2021-09-07 18:58:00

给你一个字符串数组，请你将字母异位词组合在一起。可以按任意顺序返回结果列表。
字母异位词是由重新排列源单词的字母得到的一个新单词，所有源单词中的字母都恰好只用一次。

示例 1:

输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出: [["bat"],["nat","tan"],["ate","eat","tea"]]

示例 2:

输入: strs = [""]
输出: [[""]]

示例 3:

输入: strs = ["a"]
输出: [["a"]]

提示：

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

解1 2021/9/7 O(n)

from collections import defaultdict

def groupAnagrams(strs: list) -> list:
    # 题目没有提，重复的情况，比如nnt，[nnt,ntn,tnn]，貌似是这样
    # 对于某个单词来说，用dict记录每个字母的数量，能匹配上，就算是：字母异位词
    res=[]
    # 用来存所有的dict
    ld=[]
    for s in strs:
        d=defaultdict(str)
        d_exist=False
        for c in s:
            if d.get(c): d[c]+=1
            else: d[c]=1
        # 找有没有现有的dict对的上的，没有就新增一个
        for i,x in enumerate(ld):
            # dict的比较，不论是普通的还是defaultdict，都无视内部顺序，顺序不同，内容同，比较==就是True
            if d==x:
                res[i].append(s)
                d_exist=True
                break
        if d_exist==False:
            ld.append(d)
            res.append([s])
    return res

if __name__ == '__main__':
    # [["bat"],["nat","tan"],["ate","eat","tea"]]
    strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
    print(groupAnagrams(strs))
    # [[""]]
    strs = [""]
    print(groupAnagrams(strs))
    # [["a"]]
    strs = ["a"]
    print(groupAnagrams(strs))
    strs = ["eat", "tta", "att", "ate", "nat", "bat"]
    print(groupAnagrams(strs))

Leetcode - 49. 字母异位词分组_技术