【探索-中級演算法】字謎分組
阿新 • • 發佈:2018-11-03
解題的關鍵思想就是將字串建立起無關於字母順序,只關於字母個數的唯一對映關係,這樣 ,對於同一組符合要求的字串,則它們的對映都是一樣的,以此作為分組的依據。
具體的解題方法,暫時想到了兩種。
1、將字串對映為字母+數字
的組合,如 a1b2c3
,這樣是符合上述條件的。
public static List<List<String>> groupAnagrams(String[] strs) {
if (strs==null||strs.length<1) return null;
List<List<String> > lists = new ArrayList<>();
//map 用於儲存對映,key 代表對映關係,value 代表在 lists 中的 index
Map<String, Integer> map = new HashMap<>();
for (String s : strs) {
int[] tmp = new int[26];
//計算字母的個數
for (int i = 0, l = s.length(); i < l; i++) {
tmp[s. charAt(i)-'a']++;
}
StringBuilder stringBuilder = new StringBuilder();
//對映為 字母+數字的組合,如 a1b2c3
for (int i = 0; i < 26; i++) {
if (tmp[i] > 0) {
char c = (char) ('a' + i);
//"" 是為了防止 c 被強制轉換為 int 型別然後與 tmp[i] 相加在一起,這樣就影響 key 值的唯一性
stringBuilder.append(""+c + tmp[i]);
}
}
String key = stringBuilder.toString();
if (map.containsKey(key)) {
lists.get(map.get(key)).add(s);
} else {
List<String> list = new ArrayList<>();
list.add(s);
map.put(key, lists.size());
lists.add(list);
}
}
return lists;
}
2、第二種思路就是將字串內的字母進行排序,那麼相同組合的排序後的結果肯定是一樣,同樣符合上述條件。
摘錄自 LeetCode 答案集。
public List<List<String>> groupAnagrams(String[] strs) {
Map<String, List<String>> map = new HashMap<>();
for (String str : strs) {
char[] chars = str.toCharArray();
Arrays.sort(chars);
String key = new String(chars);
if (!map.containsKey(key)) {
map.put(key, new ArrayList<>());
}
map.get(key).add(str);
}
return new ArrayList<>(map.values());
}