题目:给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
示例 1:
输入: strs = [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出: [[“bat”],[“nat”,“tan”],[“ate”,“eat”,“tea”]]
示例 2:
输入: strs = [“”]
输出: [[“”]]
示例 3:
输入: strs = [“a”]
输出: [[“a”]]
解题思路:
创建一个哈希表 map,用于存储分组后的字母异位词。这个哈希表的键是字符串的字母计数数组的字符串表示,值是一个列表,包含具有相同字母异位词的字符串。
遍历输入的字符串数组 strs,对于数组中的每个字符串 str 执行以下步骤:
a. 调用 getKey 方法将字符串 str 转换为字母计数数组的字符串表示,该字符串表示将用作字母异位词的分组标识符。
b. 如果哈希表 map 中不存在键 key,则创建一个新的列表,并将其与键 key 关联起来。这个列表将用于存储字母异位词分组。
c. 将当前字符串 str 添加到与键 key 关联的列表中。
最后,通过返回哈希表 map 的所有值(即分好组的字母异位词列表)来返回最终结果。
在 main 方法中,创建一个 Solution 类的实例,然后使用示例输入数据进行测试。
辅助函数 getKey 负责将字符串转换为字母计数数组的字符串表示。它执行以下步骤:
创建一个长度为 26 的整数数组 count,用于记录字符串中每个字符的出现次数。遍历字符串的每个字符,对于每个字符 c,将对应的计数值 count[c - ‘a’] 增加 1。创建一个 StringBuilder 对象 key 用于构建字符串表示。
遍历 count 数组的每个元素,将字符 ‘#’ 后跟计数值添加到 key 中。
返回 key 的字符串表示。
这段代码的主要思路是使用哈希表来高效地将字母异位词分组。通过将具有相同字母计数的字符串映射到相同的键,可以有效地将它们分组在一起。这是一个时间复杂度为 O(n * k * log(k)) 的解决方案,其中 n 是输入字符串数组的长度,k 是字符串的平均长度。
public class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
// 创建一个哈希表,用于分组字母异位词
// 哈希表的键是字母计数数组的字符串表示,值是分组的字母异位词列表
// 这里不引用库,自己实现一个简单的哈希表
Map<String, List<String>> map = new HashMap<>();
// 遍历输入的字符串数组,将字母异位词分组
for (String str : strs) {
// 将字符串转换为字母计数数组的字符串表示
String key = getKey(str);
// 如果哈希表中不存在该标识符,则创建一个新列表
if (!map.containsKey(key)) {
map.put(key, new ArrayList<>());
}
// 将当前字符串添加到对应的列表中
map.get(key).add(str);
}
// 返回哈希表中的所有值,即分好组的字母异位词
return new ArrayList<>(map.values());
}
// 辅助函数:将字符串转换为字母计数数组的字符串表示
private String getKey(String str) {
int[] count = new int[26];
for (char c : str.toCharArray()) {
count[c - 'a']++;
}
StringBuilder key = new StringBuilder();
for (int i : count) {
key.append('#').append(i);
}
return key.toString();
}
public static void main(String[] args) {
Solution solution = new Solution();
String[] strs = {"eat", "tea", "tan", "ate", "nat", "bat"};
List<List<String>> result = solution.groupAnagrams(strs);
System.out.println(result);
}
}