题目:
这题牛客网的题目我感觉出的有问题,如果有两三组呢,怎么能是ArrayList<String>返回值
解题思路:
由于互为字母异位词的两个字符串包含的字母相同,因此对两个字符串分别进行排序之后得到的字符串一定是相同的,故可以将排序之后的字符串作为哈希表的键。
代码:
import java.util.*;
public class LC101 {
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) {
LC101 lc = new LC101();
String[] arr = new String[]{"eat", "tea", "tan", "ate", "nat", "bat"};
List<List<String>> res = lc.groupAnagrams(arr);
for (int i = 0; i < res.size(); i++) {
System.out.println(res.get(i));
}
}
}