中级算法-字母异位词分组

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

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

比如：

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

开始做题之前，我瞟了一眼相关标签：

得到这样的提示之后，我很快就想明白了这题的解题思路。
1，首先，需要遍历整个字符串数组。

for (String s: strs) {
	...
}

2，需要对数组中的每一个字符串进行比对。我采取的办法是：将字符串打散为char数组然后排序、重新组合，得到排序后的结果。将这个有序的字符串作为分类依据。

private String sort(String str) {
    char[] splitStr = new char[str.length()];
    str.getChars(0, str.length(), splitStr, 0);
    Arrays.sort(splitStr);
    
    return new String(splitStr);
}

3，储存分类，我采用了哈希表存储所有分类。当哈希表中存在于某个字符串相匹配的分类（List）时。将这个字符串添加到对应列表。如果哈希表中没有相匹配的分类，则建立一个新的分类并添加这个字符串。

if (res.containsKey(sorted)) {
    res.get(sorted).add(s);
} else {
    List<String> arr = new ArrayList();
    arr.add(s);
    res.put(sorted, arr);
}

4，最后，以List<List>的形式返回。

完整代码：

class Solution {
    private String sort(String str) {
        char[] splitStr = new char[str.length()];
        str.getChars(0, str.length(), splitStr, 0);
        Arrays.sort(splitStr);
        
        return new String(splitStr);
    }

    public List<List<String>> groupAnagrams(String[] strs) {
        Map<String, List<String>> res = new HashMap();
        
        String sorted = sort(strs[0]);

        res.put(sorted, new ArrayList());

        for (String s: strs) {
            sorted = sort(s);
            
            if (res.containsKey(sorted)) {
                res.get(sorted).add(s);
            } else {
                List<String> arr = new ArrayList();
                arr.add(s);
                res.put(sorted, arr);
            }
        }

        return new ArrayList<List<String>>(res.values());
    }
}