leetcode 49. 字谜分组

给定一个字符串数组,将相同字谜组合在一起。(字谜是指颠倒字母顺序而成的字)

例如,给定 ["eat", "tea", "tan", "ate", "nat", "bat"],返回:

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

 

注意:所有的输入都是小写的。


咦,拿map / unordered_map + multiset 维护一下就好了。。。

vector> groupAnagrams(vector& strs) {
    map, int> mp;
    int num = 0;
    vector> res;
    for (int i = 0; i < strs.size(); ++i) {
        multiset ts;
        for (int j = 0; j < strs[i].size(); ++j)
            ts.insert(static_cast(strs[i][j]));
        if (mp.find(ts) == mp.end()) {
            mp.insert({ts, num});
            res.push_back(vector());
            res[num++].push_back(strs[i]);
        } else {
            res[mp.find(ts)->second].push_back(strs[i]);
        }
    }
    return res;
}

你可能感兴趣的:(leetcode)