leetcode49. 字母异位词分组

题目

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

示例:

输入: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”],
输出: [[“ate”,“eat”,“tea”], [“nat”,“tan”], [“bat”] ]
说明：
所有输入均为小写字母。 不考虑答案输出的顺序。

解题思路

学了一下别人使用26个素数对应26个字母，用素数的乘积来表示单词具有一样的字母，然后使用一个map保存对应字母所在res中的list的下标，每个单词都查找map中有没有，有的话则取出res中的list然后添加，没有则创建一个list然后用map记录下标，再放进res。

代码如下

class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
        //26个素数对应26个字母
		int[] primeNumber = {2,3,5,7,11,13,17,19,23,29,
				31,37,41,43,47,53,59,61,67,71,73,79,83,89,97,101};
		//保存字母所在res中的下标
        Map<Integer, Integer> map = new HashMap<>();
        List<List<String>> res = new ArrayList<>();
        int index = 0;//list计数器
        for(int i = 0; i < strs.length; i++) {
        	int ch = 1;
        	for(int j = 0; j < strs[i].length(); j++) 
        		//获取该单词所有字母对应素数的乘积
        		ch *= primeNumber[strs[i].charAt(j)-'a'];
        	if(map.containsKey(ch)) 
        		//如果已经存在字母异位词，则加入其中
        		res.get(map.get(ch)).add(strs[i]);
        	else {
        		//没有则创建一个list加入其中并在map记录下其下标
        		map.put(ch, index++);
        		List<String> list = new ArrayList<>();
        		list.add(strs[i]);
        		res.add(list);
        	}
        }
        return res;
    }
}

提交结果

成功
显示详情
执行用时 : 22 ms, 在Group Anagrams的Java提交中击败了85.81% 的用户
内存消耗 : 52.1 MB, 在Group Anagrams的Java提交中击败了73.13% 的用户