面试题 10.02. 变位词组

524 阅读1分钟

编写一种方法,对字符串数组进行排序,将所有变位词组合在一起。变位词是指字母相同,但排列不同的字符串。

注意:本题相对原题稍作修改

示例:

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

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

image.png

解题思路

因为变位词是指字母相同,但排列不同的字符串。因此我们可以对字符串进行排序,将字符串按字典序排列,而变位词的字母是相同的,因此排序以后的结果对于所有变位词应该是相同的。

代码

class Solution {
    public List<List<String>> groupAnagrams(String[] strs) {

        Map<String,List<String>> map=new HashMap<>();
        for (String str : strs) {
            char[] c=str.toCharArray();
            Arrays.sort(c);
            String s = new String(c);
            if(!map.containsKey(s))
            {
                map.put(s,new ArrayList<>());
            }
            map.get(s).add(str);
        }
        List<List<String>> res=new ArrayList<>();
        for (Map.Entry<String, List<String>> entry : map.entrySet()) {
            res.add(entry.getValue());
        }
        return res;
        
    }
}