【LeetCode】49. Group Anagrams

题目：

Given an array of strings, group anagrams together.

For example, given: ["eat", "tea", "tan", "ate", "nat", "bat"], Return:

[
  ["ate", "eat","tea"],
  ["nat","tan"],
  ["bat"]
]

Note:

1. For the return value, each inner list's elements must follow the lexicographic order.
2. All inputs will be in lower-case.

提示：

这道题要把所有字母组成相同的单词归为一类。因此我们可以把每个字母都进行排序，然后利用一个hash_map保存。即以排序后的结果作为键，map的值可以是一个set，把排序前的结果插入到set当中。由于set的底层实现利用了平衡二叉搜索树，所以插入以后的元素是已经排好序的。这样归类就完成了。

代码：

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>> res;
        if (strs.empty()) {
            return res;
        }
        unordered_map<string, multiset<string>> um;
        for (string str : strs) {
            string tmp = str;
            sort(tmp.begin(), tmp.end());
            um[tmp].insert(str);
        }
        for (auto m : um) {
            vector<string> sol(m.second.begin(), m.second.end());
            res.push_back(sol);
        }
        return res;
    }
};

实际上，由于对单词排序时，题目已经限定了单词只可能是26个小写字母组成的，所以我们可以使用计数排序进一步加快算法的速度（排序部分速度从O(nlogn)变为O(n)），代码如下：

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>> res;
        if (strs.empty()) {
            return res;
        }
        unordered_map<string, multiset<string>> um;
        for (string str : strs) {
            string tmp = strSort(str);
            um[tmp].insert(str);
        }
        for (auto m : um) {
            vector<string> sol(m.second.begin(), m.second.end());
            res.push_back(sol);
        }
        return res;
    }
    
    string strSort(string s) {
        vector<int> count(26, 0);
        for (int i = 0; i < s.length(); ++i) {
            ++count[s[i] - 'a'];
        }
        string res = "";
        for (int i = 0; i < 26; ++i) {
            while (count[i]--) {
                res += ('a' + i);
            }
        }
        return res;
    }
};