Given an array of strings, return all groups of strings that are anagrams.
Note: All inputs will be in lower-case.
原文链接:http://www.cnblogs.com/AnnieKim/archive/2013/04/25/3041982.html
思路:
用map<string, int>记录排序后的字符串以及首次出现的位置。
1. 从strs的第一个元素开始遍历,首先对元素进行排序得到s;
2. 在map里查找s;
3. 若不存在,将s以及该元素的下标存入map<string ,int>;
4. 若存在,首先将第一次出现s时的原始字符串存入结果res,即strs[map[s]],并将map[s]设置为-1(防止下次再存),再将该字符串本身存入结果res;
5. 重复以上1-4步,直到遍历结束。
代码:
class Solution {
public:
vector<string> anagrams(vector<string> &strs) {
vector<string> res;
if(strs.size()<=1) return res;
map<string,int> anagram;
for(int i=0;i<strs.size();i++){
string s=strs[i];
sort(s.begin(),s.end());
if(anagram.find(s) == anagram.end()){
anagram[s]=i;
}else{
if(anagram[s]>=0){
res.push_back(strs[anagram[s]]);
anagram[s]=-1;
}
res.push_back(strs[i]);
}
}
return res;
}
};