方法一:排序
输入: strs = [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出: [[“bat”],[“nat”,“tan”],[“ate”,“eat”,“tea”]]
从样例我们可以直观看到,所谓异位词分组,就是单词的字母相同,即排序后的单词相同。我们可以使用哈希表存储异位词,排序后的单词作为哈希表的键,一组异位词即位哈希表的值。
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
Map<String, List<String>> map = new HashMap<>();
for(String str : strs){
char[] array = str.toCharArray();
Arrays.sort(array);
String key = new String(array);
List<String> list = map.getOrDefault(key,new ArrayList<String>());
list.add(str);
map.put(key,list);
}
return new ArrayList<List<String>>(map.values());
}
}
时间复杂度:
对一个array排序的时间是O(klogk),k为字符串的最大长度。
外层遍历n个,所以时间复杂度是O(nklogk)
方法二:计数
排序相当于统计每个字母的出现次数是否一样,所以我们也可以直接统计每个字母的出现次数,将字符和出现次数作为哈希表的键,一组异位词作为哈希表的值。
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
Map<String, List<String>> map = new HashMap<>();
for(String str : strs){
int[] counts = new int[26];
int length = str.length();
for(int i = 0; i < length; i++){
counts[str.charAt(i) - 'a']++;
}
StringBuffer sb = new StringBuffer();
for(int i = 0;i < 26; i++){
if(counts[i] > 0){
sb.append((char)'a' + i);
sb.append(counts[i]);
}
}
String key = sb.toString();
List<String> list = map.getOrDefault(key, new ArrayList<String>());
list.add(str);
map.put(key,list);
}
return new ArrayList<List<String>>(map.values());
}
}
在拼接sb时,我一开始用的两层循环,来达到和方法一一样的键,但其实没必要,直接拼上数字有一样的效果,且降低了时间复杂度
时间复杂度:
O(k)遍历一个字符串,k是最大长度。
O(∣Σ∣),∣Σ∣是字符集,该题即为26
有n个字符串,所以最后时间复杂度是O(n(k+∣Σ∣))