给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
输入: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”],
输出:
[
[“ate”,“eat”,“tea”],
[“nat”,“tan”],
[“bat”]
]
说明:
所有输入均为小写字母。
不考虑答案输出的顺序。
思路很清晰,不用多解释,代码如下:
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
if (strs.length == 0)
return new ArrayList();
Map<String,List> map = new HashMap<>();//存放字符完全相同的字符组合
for (String s : strs) {
char[] chs = s.toCharArray();//将字符串转为字符数组
Arrays.sort(chs);//将字符数组排序,以便于判断是否相同
String temp = String.valueOf(chs);//key值
if (!map.containsKey(temp))//不存在key值则加入
map.put(temp,new ArrayList());
map.get(temp).add(s);//将当前字符串和对应key关联
}
return new ArrayList(map.values());//拿到所有的values
}
}
根据上述代码,还有一丝改进空间,代码如下:
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
if (strs.length == 0) {
return new ArrayList<>();
}
Map<String, ArrayList<String>> map = new HashMap<>();
for (String str : strs) {
char[] chars = str.toCharArray();
Arrays.sort(chars);
String key = String.valueOf(chars);
map.putIfAbsent(key, new ArrayList<>());
//这个方法可自行上网搜索
map.get(key).add(str);
}
return new ArrayList(map.values());
}
}