欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode:49. 字母异位词分组

程序员文章站 2022-07-12 12:49:45
...

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

示例:

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

说明:

  • 所有输入均为小写字母。
  • 不考虑答案输出的顺序。
class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {
        List<List<String>> listAll=new ArrayList<List<String>>();
        
        if(strs.length==0)
        	return listAll;
 
        Map<Double,Integer> map=new HashMap<Double,Integer>();
        int a[]={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};       
                
        for(int i=0;i<strs.length;i++){
        	double tmp=1;
        	for(int j=0;j<strs[i].length();j++){
        		tmp=tmp*a[strs[i].charAt(j)-'a'];
        	}
        	if(map.containsKey(tmp)){  //如果已经出现了同样乘积的字符串
        		listAll.get(map.get(tmp)).add(strs[i]);  //直接存储

    		}else{  //如果第一次出现不一样的字符串
    			map.put(tmp,listAll.size()); //map中存档
    			List<String> list=new ArrayList<String>();  //新建List存入listall
    			list.add(strs[i]);
    			listAll.add(list);
    		}
        	
        }
        
	        
        return listAll;
    }
}

思路借鉴了评论区的大佬的思路:用质数表示26个字母,把字符串的各个字母相乘,这样可保证字母异位词的乘积必定是相等的。用map存储,KEY表示存储乘积,value表示该乘积对应的listAll中的下标,得到下标即可知道把该字符串存储到listAll中第几个List中。