【题目】
Given an array of strings, return all groups of strings that are anagrams.
Note: All inputs will be in lower-case.
【解析】题意:给定一个字符串数组,返回所有是“换位词”的字符串。
所谓“换位词/变位词”就是包含相同字母,但字母顺序可能不同的字符串。比如“abc", "bca", "cab", "acb", "bac", "cba"都互为换位词。
很容易想到,这些换位词如果按字母序排序后都是”abc“,所以可以对原数组中每个字符串自身进行排序,然后出现过两次以上的就是我们要找的字符串。
需要注意的是,要保持原字符串数组不变,那么就需要一个同样大小的数组来存储排序后的每个字符串。
接下来的问题是,如何找出出现两次以上的字符串对应的原字符串,这个用代码比较难以实现,两层for循环的暴力解决会超时。
我这里用了一个比较笨的方法实现:再用一个同样大小的数组标记相应下标位置的原字符串是否为“换位词”,这样需要一边遍历,然后再一边遍历把相应位置的原字符串添加到结果集中。时间复杂度为O(n)。
public class Solution {
public List<String> anagrams(String[] strs) {
if (strs.length < 1) return null;
List<String> ans = new ArrayList<String>();
String[] strs2 = new String[strs.length];
for (int i = 0; i < strs.length; i++) {
char[] chs = strs[i].toCharArray();
Arrays.sort(chs);
strs2[i] = String.valueOf(chs);
}
boolean[] flags = new boolean[strs.length];
Arrays.fill(flags, false);
HashMap<String, Integer> map = new HashMap<String, Integer>();
for (int i = 0; i < strs2.length; i++) {
if (map.containsKey(strs2[i])) {
int last = map.put(strs2[i], i);
flags[last] = true;
flags[i] = true;
} else {
map.put(strs2[i], i);
}
}
for (int i = 0; i < strs.length; i++) {
if (flags[i]) {
ans.add(strs[i]);
}
}
return ans;
}
}