问题描述
Given an array of strings, group anagrams together.
Example:
Input: ["eat", "tea", "tan", "ate", "nat", "bat"]
,
Output:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
Note:
- All inputs will be in lowercase.
- The order of your output does not matter.
分析与设计
因为会使用到一些字典、列表和字符串的操作,所以把此题记录下来。可以看到我们的目标是把元素相同的字符串归为一组,这个时候可以想象到每组元素都有一个共同的特征,那么这个共同特征就可以作为字典的索引,帮助我们把他分类。贴上代码,逐行分析:
class Solution:
def groupAnagrams(self, strs):
"""
:type strs: List[str]
:rtype: List[List[str]]
"""
res = []
mdict = {}
for i in range(len(strs)):
tmp = strs[i]
sotmp = list(strs[i])
sotmp.sort()
if ''.join(sotmp) in mdict.keys():
mdict[''.join(sotmp)] += [tmp]
else:
mdict[''.join(sotmp)] = [tmp]
{res.append(value) for key,value in mdict.items()}
return res
首先我们要判断字符串中元素是否相同,并且找一个成为索引的依据,可以想到字符排序是一个简单直接的办法,而字符串不能(或者我不知道)直接排序,所以先将其转换为List,同时我们不希望丢失掉这个字符串,所以新建立一个变量用来存储。
剩下的逻辑其实很简单,主要是有几个列表字典字符串的操作:
- "".join(List)操作,用于将列表中的元素(str) 以指定的字符 连接生成一个新的字符串。
- List+List,用于将两个列表结合,组成一个新的列表
- {res.append(value) for key,value in mdict.items()},key和value分别代表字典中的键和值,我们要取value作为最后的结果,使用List.append()操作,即为给列表添加新的对象,与List+List不同的是,前者并不结合为一体。