给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
输入: ["eat", "tea", "tan", "ate", "nat", "bat"]
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
说明:
- 所有输入均为小写字母。
- 不考虑答案输出的顺序。
package Solution49;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
class Solution3 {
public List<List<String>> groupAnagrams(String[] strs) {
List<List<String>> out = new ArrayList<List<String>>();
HashSet<String> hs = new HashSet<String>();
for (int i = 0; i < strs.length; i++) {
char[] temp = strs[i].toCharArray();
Arrays.sort(temp);
hs.add(Arrays.toString(temp));
}
for (String ele : hs) {
List<String> templist = new ArrayList<String>();
for (int i = 0; i < strs.length; i++) {
char[] temp = strs[i].toCharArray();
Arrays.sort(temp);
if (Arrays.toString(temp).equals(ele)) {
templist.add(strs[i]);
}
}
out.add(templist);
}
return out;
}
public static void main(String[] args) {
Solution3 sol = new Solution3();
String[] strs = { "eat", "tea", "tan", "ate", "nat", "bat" };
System.out.println(sol.groupAnagrams(strs));
}
}
超时
官方答案思路一样但写的很不一样
package Solution49;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
Map<String, List<String>> map = new HashMap<String, List<String>>();
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());
}
public static void main(String[] args) {
Solution sol = new Solution();
String[] strs = { "eat", "tea", "tan", "ate", "nat", "bat" };
System.out.println(sol.groupAnagrams(strs));
}
}