字母异位词分组
给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
示例 1:
输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
示例 2:
输入: strs = [""]
输出: [[""]]
示例 3:
输入: strs = ["a"]
输出: [["a"]]
提示:
1 <= strs.length <= 104
0 <= strs[i].length <= 100
strs[i] 仅包含小写字母
thought: 题目核心是找字母相等的字符串,所以可以通过字母排序以后进行对比。
代码1(并未成功):
由于互为字母异位词的两个字符串包含的字母相同,因此对两个字符串分别进行排序之后得到的字符串一定是相同的,故可以将排序之后的字符串作为哈希表的键。
# 第一种并未成功的算法:这题采用选择,然后再排序,再对比。时间复杂度n^3logn
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
List<List<String>> result = new ArrayList<>();
for (int i = 0;i<strs.length;i++){
String str = strs[i];
char[] value1 = str.toCharArray();
Arrays.sort(value1);
List<String> res = new ArrayList<>();
for ( int j = i+1;j<strs.length;j++){
String str2 = strs[j];
char[] value2 = str2.toCharArray();
Arrays.sort(value2);
if(Arrays.toString(value1).equals(Arrays.toString(value2))){
if (res.size() == 0){
res.add(str);
}
res.add(str2);
}
}
result.add(res);
}
return result;
}
}
代码2(吸取代码1的失误,查看官方解读)
thought:同样采用排序的方法去比较字符,但是这里使用排序后的字符串作为key,来找到异位词。时间复杂度o(nlogn)。使用了map类型的getOrDefault方法来获取每个数组应存值。
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
Map<String,List<String>> map = new HashMap<String,List<String>>();
for(String str: strs){
char[] value = str.toCharArray();
Arrays.sort(value);
String key = Arrays.toString(value);
List<String> res = map.getOrDefault(key, new ArrayList<String>());
res.add(str);
map.put(key,res);
}
return new ArrayList<List<String>>(map.values());
}
}
python方法:
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
mp = collections.defaultdict(list)
for st in strs:
key = "".join(sorted(st))
mp[key].append(st)
return list(mp.values())