题目:
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。 示例: 输入: ["eat", "tea", "tan", "ate", "nat", "bat"], 输出: [ ["ate","eat","tea"], ["nat","tan"], ["bat"] ] 说明: 所有输入均为小写字母。 不考虑答案输出的顺序。
解法:
class Solution { public: vector<vector<string>> groupAnagrams(vector<string>& strs) { map<string,set<int>> str_map; vector<vector<string>> res; for(int i = 0;i < strs.size();++i){ string tmp = strs[i]; sort(tmp.begin(),tmp.end()); if(str_map.find(tmp)!=str_map.end()){ str_map[tmp].insert(i); }else{ set<int> t; t.insert(i); str_map.insert(make_pair(tmp,t)); } } map<string,set<int>>::iterator it = str_map.begin(); for(;it!=str_map.end();++it){ vector<string> str_vec; set<int>::iterator set_it = (it->second).begin(); for(;set_it!=(it->second).end();++set_it){ str_vec.push_back(strs[*set_it]); } res.push_back(str_vec); } return res; } };