题目:
Given an array of strings, return all groups of strings that are anagrams.
Note: All inputs will be in lower-case.
Hash Table String链接: 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