LeetCode Solutions
49. Group Anagrams
Time: $O(nk\log k)$, where $n = |\texttt{strs}|$ and $k = |\texttt{strs[i]}|$ Space: $O(nk)$
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
vector<vector<string>> ans;
unordered_map<string, vector<string>> keyToAnagrams;
for (const string& str : strs) {
string key = str;
sort(begin(key), end(key));
keyToAnagrams[key].push_back(str);
}
for (const auto& [_, anagrams] : keyToAnagrams)
ans.push_back(anagrams);
return ans;
}
};
class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
Map<String, List<String>> keyToAnagrams = new HashMap<>();
for (final String str : strs) {
char[] chars = str.toCharArray();
Arrays.sort(chars);
String key = String.valueOf(chars);
keyToAnagrams.computeIfAbsent(key, k -> new ArrayList<>()).add(str);
}
return new ArrayList<>(keyToAnagrams.values());
}
}
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
dict = defaultdict(list)
for str in strs:
key = ''.join(sorted(str))
dict[key].append(str)
return dict.values()