leetcode刷题之第49题：字母异位词分组

题目描述

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

示例:

输入: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出:
[
[“ate”,“eat”,“tea”],
[“nat”,“tan”],
[“bat”]
]
说明：

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

思路

直观的想法是比较两个单词各字母的ASCII码，但实现起来较为麻烦而且速度慢。可以先将单词的字母从小到大排序，然后判断排序后的单词是否相等，相等则为同一组。

代码

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>>  res;
        int sub=0;
        string tmp;
        unordered_map<string,int> work;
        for(int i=0;i<strs.size();i++)
        {
            tmp=strs[i];
            sort(tmp.begin(),tmp.end());
            if(work.count(tmp))
            {
                res[work[tmp]].push_back(strs[i]);
            }
            else
            {
                vector<string> vec(1,strs[i]);
                res.push_back(vec);
                work[tmp]=sub++;
            }
        }
        return  res;
    }
};