[数据结构与算法]Leetcode T49: 字母异位词分组

题目描述

给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

字母异位词 是由重新排列源单词的字母得到的一个新单词，所有源单词中的字母通常恰好只用一次。

示例 1:

输入: strs = [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出: [[“bat”],[“nat”,“tan”],[“ate”,“eat”,“tea”]]

示例 2:

输入: strs = [“”]
输出: [[“”]]

示例 3:

输入: strs = [“a”]
输出: [[“a”]]

提示：

• 1 <= strs.length <= 104
• 0 <= strs[i].length <= 100
• strs[i] 仅包含小写字母

思路

如果两个字符串再根据字符进行排序后，得到的字符串相等，那么这两个字符串为字母异位词。
使用HashMap，提高存取效率。

代码

	Map<String, List<Integer>> map = new HashMap<String, List<Integer>>();
	
    public List<List<String>> groupAnagrams(String[] strs) {
    	for(int i = 0; i < strs.length; i++) {
    		char[] chs = strs[i].toCharArray();
    		Arrays.sort(chs);
    		if(map.containsKey(new String(chs))) {
    			map.get(new String(chs)).add(i);
    		} else {
    			List<Integer> lis = new ArrayList<Integer>();
    			lis.add(i);
    			map.put(new String(chs), lis);
    		}
    	}
    	List<List<String>> res = new ArrayList<List<String>>();
    	for(List<Integer> lis: map.values()) {
    		List<String> tmp = new ArrayList<String>();
    		for(int ind: lis) tmp.add(strs[ind]);
    		res.add(tmp);
    	}
    	return res;
    }