LeetCode 字谜分组

给定一个字符串数组,将字母异位词组合在一起。
字母异位词指字母相同,但排列不同的字符串。

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
  ["ate","eat","tea"],
  ["nat","tan"],
  ["bat"]
]
说明:

所有输入均为小写字母。
不考虑答案输出的顺序。
解法一:

这道题考察的就是如何判断两个字符串是否是字母异位词,这里有一个方法,如果把字母异位词重新排序,会得到相同的结果,所以重新排序可以判断两个字符串是否互为字母异位词。由于字母异位词重新排序后都会得到相同的字符串,所以我们可以定义一个 HashMap>,以排序后的字符串作为 key,然后将所有字母异位词保存到该 key 对应的 ArrayList 中,最后再返回一个包含所有 ArrayList 的 List> 即可。

    public List> groupAnagrams(String[] strs) {
        HashMap> hashMap = new HashMap<>();

        for (String str : strs) {
            char[] chars = str.toCharArray();
            Arrays.sort(chars);
            String key = new String(chars);

            if (!hashMap.containsKey(key)) {
                hashMap.put(key, new ArrayList<>());
            }
            hashMap.get(key).add(str);
        }

        return new ArrayList<>(hashMap.values());
    }
解法二:

如果不想用排序,我们还可以用一个大小为 26 的 int 数组来统计每个单词中每个字符出现的次数,然后将 int 数组转为一个唯一的字符串作为 key。因为字母异位词的 key 都是一样的,所以就可以将字符串存入自己 key 对应的 ArrayList 中,最后再返回一个包含所有 ArrayList 的 List> 即可。


    public List> groupAnagrams2(String[] strs) {

        HashMap> hashMap = new HashMap<>();

        StringBuilder stringBuilder = new StringBuilder();
        for (int i = 0; i < strs.length; i++) {

            stringBuilder.delete(0, stringBuilder.length());
            int[] count = new int[26];

            for (int j = 0; j < strs[i].length(); j++) {
                count[strs[i].charAt(j) - 'a']++;
            }

            for (int k = 0; k < 26; k++) {
                stringBuilder.append(count[k] + '/');
            }

            if (!hashMap.containsKey(stringBuilder.toString())) {
                hashMap.put(stringBuilder.toString(), new ArrayList<>());
            }
            hashMap.get(stringBuilder.toString()).add(strs[i]);
        }

        return new ArrayList<>(hashMap.values());
    }

你可能感兴趣的:(LeetCode 字谜分组)