题目大意
给你一个字符串数组,请你将字母异位词组合在一起。可以按任意顺序返回结果列表。
字母异位词是由重新排列源单词的所有字母得到的一个新单词。
思路分析
示例 1:
输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
上面的示例是什么意思那?就是把单词列表中字母组成相同的放在一个列表中。在输出的子列表中包含的单词,其字母构成是一样的。
既然单词的字母之间不要求有序,那么只需要对每个单词进行排序,借助一个字典|Map对所有单词映射即可。
代码
Java版本
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
Map<String, List<String>> map = new HashMap<>();
for(String str:strs){
char [] chars = str.toCharArray();
Arrays.sort(chars);
String key = new String(chars);
List<String> value = map.getOrDefault(key,new ArrayList<String>());
value.add(str);
map.put(key,value);
}
return new ArrayList<List<String>>(map.values());
}
}
Python版本
class Solution(object):
def groupAnagrams(self, strs):
"""
:type strs: List[str]
:rtype: List[List[str]]
"""
res = {}
for item in strs:
list_item = list(item)
list_item = sorted(list_item)
str_item = "".join(list_item)
if str_item not in res:
res[str_item] = []
res[str_item].append(item)
out = []
for k, v in res.items():
out.append(v)
return out