https://oj.leetcode.com/problems/anagrams/
Given an array of strings, return all groups of strings that are anagrams.
Note: All inputs will be in lower-case.
public List<String> anagrams(String[] strs)
问题分析:anagram实际上就是同构词,也就是字符串的字符顺序可以不同,但是字符的数量和内容相同,也就是aabbcc和abcabc就是一对anagram,它们有相同的a,b,c的数量。
下面我就给两种做法。
第一种的做法是O(nmlogm), 空间O(n)的HashMap的做法,n是数组的长度,m是字符串数组里字符串的平均长度。哈希表里面,键值是排序好的字符串,因为所有anagram的排序过的字符串都是一样的。所以以此为键,然后哈希表的值域是一个List<String>,将同一键值的串在一起。然后再遍历哈希表,将值域里的List串在一起。
public List<String> anagrams(String[] strs) {
HashMap<String, List<String>> cached_map = new HashMap<String, List<String>>();
for(String s : strs){
char[] str_to_char = s.toCharArray();
Arrays.sort(str_to_char);
String str = new String(str_to_char);
if(cached_map.containsKey(str)){
List<String> cur_res = cached_map.get(str);
cur_res.add(s);
cached_map.put(str, cur_res);
}else{
List<String> cur_res = new LinkedList<String>();
cur_res.add(s);
cached_map.put(str, cur_res);
}
}
List<String> res = new LinkedList<String>();
for(String s : cached_map.keySet()){
List<String> cur_res = cached_map.get(s);
if(cur_res.size() > 1){
for(String str : cur_res)
res.add(str);
}
}
return res;
}
这个代码比较明晰和简短,个人推荐这种做法。
第二种做法时间上是O(nlogn * mlogm),但不占空间。实际上基本理念和上一个做法是一样的:所有anagram的排序过的字符串都是一样的。上一做法里面我们只排序每一个字符串。这一种做法则是排序整个字符串数组,比较器的compare方法则是将两个需要比较的字符串先排序内部内容,然后用String自带的compareTo进行比较。这种做法的特点在于经过排序的字符串数组,作为anagram的那些单词是会排在一起的。然后前后比较即可。
public List<String> anagrams(String[] strs) {
Arrays.sort(strs, new StringComparator());
LinkedList<String> res = new LinkedList<String>();
if(strs.length == 0)
return res;
String tmp = strs[0];
boolean found = false;
for(int i = 1; i < strs.length; i++){
if(compare(tmp, strs[i]) == 0){
if(!found)
res.add(tmp);
found = true;
res.add(strs[i]);
}else{
found = false;
tmp = strs[i];
}
}
return res;
}
public int compare(String s1, String s2){
char[] sort_1 = s1.toCharArray();
char[] sort_2 = s2.toCharArray();
Arrays.sort(sort_1);
Arrays.sort(sort_2);
String sorted_1 = new String(sort_1);
String sorted_2 = new String(sort_2);
return sorted_1.compareTo(sorted_2);
}
public class StringComparator implements Comparator<String>{
@Override
public int compare(String s1, String s2){
char[] sort_1 = s1.toCharArray();
char[] sort_2 = s2.toCharArray();
Arrays.sort(sort_1);
Arrays.sort(sort_2);
String sorted_1 = new String(sort_1);
String sorted_2 = new String(sort_2);
return sorted_1.compareTo(sorted_2);
}
}