给定一个字符串数组,将字母异位词组合在一起。
字母异位词指字母相同,但排列不同的字符串。
示例:
输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
说明:
所有输入均为小写字母。
不考虑答案输出的顺序。
解法一:
这道题考察的就是如何判断两个字符串是否是字母异位词,这里有一个方法,如果把字母异位词重新排序,会得到相同的结果,所以重新排序可以判断两个字符串是否互为字母异位词。由于字母异位词重新排序后都会得到相同的字符串,所以我们可以定义一个 HashMap
,以排序后的字符串作为 key,然后将所有字母异位词保存到该 key 对应的 ArrayList 中,最后再返回一个包含所有 ArrayList 的 List
即可。>
public List> groupAnagrams(String[] strs) {
HashMap> hashMap = new HashMap<>();
for (String str : strs) {
char[] chars = str.toCharArray();
Arrays.sort(chars);
String key = new String(chars);
if (!hashMap.containsKey(key)) {
hashMap.put(key, new ArrayList<>());
}
hashMap.get(key).add(str);
}
return new ArrayList<>(hashMap.values());
}
解法二:
如果不想用排序,我们还可以用一个大小为 26 的 int 数组来统计每个单词中每个字符出现的次数,然后将 int 数组转为一个唯一的字符串作为 key。因为字母异位词的 key 都是一样的,所以就可以将字符串存入自己 key 对应的 ArrayList 中,最后再返回一个包含所有 ArrayList 的 List
即可。>
public List> groupAnagrams2(String[] strs) {
HashMap> hashMap = new HashMap<>();
StringBuilder stringBuilder = new StringBuilder();
for (int i = 0; i < strs.length; i++) {
stringBuilder.delete(0, stringBuilder.length());
int[] count = new int[26];
for (int j = 0; j < strs[i].length(); j++) {
count[strs[i].charAt(j) - 'a']++;
}
for (int k = 0; k < 26; k++) {
stringBuilder.append(count[k] + '/');
}
if (!hashMap.containsKey(stringBuilder.toString())) {
hashMap.put(stringBuilder.toString(), new ArrayList<>());
}
hashMap.get(stringBuilder.toString()).add(strs[i]);
}
return new ArrayList<>(hashMap.values());
}