面试题 10.02. 变位词组-----计数
题目表述
编写一种方法,对字符串数组进行排序,将所有变位词组合在一起。变位词是指字母相同,但排列不同的字符串。
注意:本题相对原题稍作修改
示例:
示例1:
输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
Map计数
由于互为变位词的两个字符串包含的字母相同,因此两个字符串中的相同字母出现的次数一定是相同的,故可以将每个字母出现的次数使用字符串表示,作为哈希表的键。
由于字符串只包含小写字母,因此对于每个字符串,可以使用长度为 26的数组记录每个字母出现的次数。需要注意的是,在使用数组作为哈希表的键时,不同语言的支持程度不同,因此不同语言的实现方式也不同。
class Solution {
List<List<String>> res = new ArrayList<>();
public List<List<String>> groupAnagrams(String[] strs) {
Map<String, List<String>> map = new HashMap<>();
for(String str : strs){
int[] count = new int[26];
StringBuilder tmp = new StringBuilder();
for(char ch : str.toCharArray()){
count[ch - 'a'] += 1;
}
for(int i = 0; i < 26; i++){
if(count[i] > 0){
tmp.append('a' + i);
tmp.append(count[i]);
}
}
List<String> list = map.getOrDefault(tmp.toString(), new ArrayList<String>());
list.add(str);
map.put(tmp.toString(), list);
}
return new ArrayList<List<String>>(map.values());
}
}