题目:
Given an array of strings, return all groups of strings that are anagrams.
Note: All inputs will be in lower-case.
链接: http://leetcode.com/problems/anagrams/
题解:
求String数组里所有的异位构词anagram。比如"abcde"和"edcba"算是anagrams,"12345"和"54321"。结果要求返回所有的anagrams在一个ArrayList里,所以我们例子里的结果就是{"abcde", "edcba","12345","54321"}。原理是利用一个HashMap存储String key以及ArrayList<String> value。 Key是排序好的string,比如"abcde"和"edcba"排序后都是"abcde",所以他们都属于同一组anagrams。对于排序后的string,已经在HashMap里存在key的,我们把排序前的原string加入到结果里。当HashMap里存在key而且value.size()为1时,我们要把第一个未排序的string先存入结果。 当HashMap里不存在key,我们要建立这个<key,value>pair。时间复杂度主要在于要对n个string,每个string转换为array之后进行排序,java的Arrays.sort()对于数组的话是使用merge sort,所以时间复杂度是O(n * mlogm),m是每个string的平均长度。空间复杂度的话我们使用了HashMap,临时数组可以不考虑,所以主要是O(n * m)。
Time Complexity - O(n * mlogm),Space Complexity - O(n * m),where m is average length of elements in strs。
public class Solution {
public ArrayList<String> anagrams(String[] strs) {
ArrayList<String> result = new ArrayList<String>();
if(strs == null || strs.length == 0)
return result;
HashMap<String, ArrayList<String>> map = new HashMap<String, ArrayList<String>>(); for(int i = 0; i < strs.length; i++){
char[] arr = strs[i].toCharArray();
Arrays.sort(arr);
String sortedStr = String.valueOf(arr);
if(map.containsKey(sortedStr)){
if(map.get(sortedStr).size() == 1)
result.add(map.get(sortedStr).get(0));
map.get(sortedStr).add(strs[i]);
result.add(strs[i]);
} else {
ArrayList<String> list = new ArrayList<String>();
list.add(strs[i]);
map.put(sortedStr, list);
}
} return result;
}
}
更新: Group anagrams
public class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
List<List<String>> res = new ArrayList<>();
if(strs == null || strs.length == 0)
return res;
HashMap<String, ArrayList<String>> map = new HashMap<>();
Arrays.sort(strs); for(String str : strs) {
char[] arr = str.toCharArray();
Arrays.sort(arr);
String sortedStr = new String(arr);
if(!map.containsKey(sortedStr)) {
ArrayList<String> list = new ArrayList<>();
list.add(str);
map.put(sortedStr, list);
} else {
map.get(sortedStr).add(str);
}
} for(String str : map.keySet())
res.add(map.get(str)); return res;
}
}
二刷:
现在这道题已经改名叫Group Anagrams了,还加了例子。好人性化的感觉。主要思路还是和一刷一样,对每一个Strs里面的字符串,先生成排序后的新字符串key,然后根据key保存到hashmap里。最后遍历hashmap把结果放在List里。
看了discuss从wz366看到可以直接把map的values转化为结果 - return new ArrayList<List<String>>(map.values());
Given an array of strings, group anagrams together.
For example, given: ["eat", "tea", "tan", "ate", "nat", "bat"]
,
Return:
[
["ate", "eat","tea"],
["nat","tan"],
["bat"]
]
Note:
- For the return value, each inner list's elements must follow the lexicographic order.
- All inputs will be in lower-case.
Java:
Time Complexity - O(n * mlogm) , Space Complexity - O(m * n), m为字符串的平均长度
public class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
List<List<String>> res = new ArrayList<>();
if (strs == null || strs.length == 0) {
return res;
}
Map<String, List<String>> map = new HashMap<>();
for (String s : strs) {
char[] arr = s.toCharArray();
Arrays.sort(arr);
String key = new String(arr);
if (map.containsKey(key)) {
map.get(key).add(s);
} else {
List<String> tmp = new ArrayList<>();
tmp.add(s);
map.put(key, tmp);
}
} List<String> anagrams = new ArrayList<>();
for (String str : map.keySet()) {
anagrams = map.get(str);
Collections.sort(anagrams);
res.add(anagrams);
}
return res;
}
}
或者
public class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
List<List<String>> res = new ArrayList<>();
if (strs == null || strs.length == 0) {
return res;
}
Arrays.sort(strs);
Map<String, List<String>> map = new HashMap<>();
for (String s : strs) {
char[] arr = s.toCharArray();
Arrays.sort(arr);
String key = new String(arr);
if (map.containsKey(key)) {
map.get(key).add(s);
} else {
List<String> tmp = new ArrayList<>();
tmp.add(s);
map.put(key, tmp);
}
}
return new ArrayList<List<String>>(map.values());
}
}
三刷:
方法和前两刷一样。这里先对strs数组进行排序,最后可以一次性返回new ArrayList<List<String>>(map.values);
这样写速度并不快。我们也可以先不排序,计算完hashmap以后, 在遍历hashMap将其每个key对应的values加入到res的时候进行排序。这样的话速度可以增快不少。也就是说跟二刷的第一段代码差不多。
Java:
public class Solution {
public List<List<String>> groupAnagrams(String[] strs) {
List<List<String>> res = new ArrayList<>();
if (strs == null) return res;
Arrays.sort(strs);
Map<String, List<String>> map = new HashMap<>();
for (String s : strs) {
char[] sArr = s.toCharArray();
Arrays.sort(sArr);
String key = String.valueOf(sArr);
if (!map.containsKey(key)) map.put(key, new ArrayList<String>());
map.get(key).add(s);
}
return new ArrayList<>(map.values());
}
}
Reference:
https://leetcode.com/discuss/58561/share-my-short-java-solution
https://leetcode.com/discuss/51190/1-line-ruby-python-for-updated-problem