LeetCode 49-字母异位分组
给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例
输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/group-anagrams
题解
对map一直用的比较少,这里总结一下作为一个例题好了。大致就是把每个单词排序作为map中的key,value记录拥有这个key的单词应该放到那个vector中。
map加入元素为insert函数,需要加入一个pair。
map查找key对应的value可以直接将key作为下标取值,如果不存在就会直接新加进去。所以一般先用count函数,返回值为1或者0代表有没有(key已存在时不会插入新的pair),或者使用find返回迭代器,删除是erase一个迭代器。
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
vector<vector<string>> ans;
map<string, int> mp;
int n=strs.size();
int cnt=0;
for(int i=0;i<n;i++){
string tmp=strs[i];
sort(tmp.begin(),tmp.end());
if(!mp.count(tmp)){
mp.insert(make_pair(tmp,cnt));
cnt++;
vector<string> group;
group.push_back(strs[i]);
ans.push_back(group);
}
else{
int index=mp[tmp];
ans[index].push_back(strs[i]);
}
}
return ans;
}
};