題目:
這題牛客網的題目我感覺出的有問題,如果有兩三組呢,怎麼能是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));
}
}
}