个人思路总结:
首先来说说自己的思路,用两层循环,每一次内层循环都得到一个异位词vector(对每一个string都做排序,对排序后的string进行比较,如果相等的话就将其push进vector,如果不一样就跳过)。这个方法的时间复杂度为O(n^2)。因此对于最后一个变态的测试用例没有通过(超时),不过也是写出来记录一下吧。
代码如下:
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
vector<vector<string>> result;
vector<string> tmp;
string str1, str2; //临时变量,要不排序完成之后就全部变成一样的了
while(!strs.empty())
{
tmp.clear(); //清空临时vector
tmp.push_back(strs.back());
strs.pop_back();
str1 = tmp[0];
sort(str1.begin(),str1.end());
for(auto it=strs.begin();it!=strs.end();)
{
str2 = *it;
sort(str2.begin(),str2.end());
if(str1==str2)
{
tmp.push_back(*it);
it = strs.erase(it); //注意这里删除vector中某个元素的操作
}
else
it++;
}
result.push_back(tmp);
}
return result;
}
};
大神思路:
维护一个映射 ans : {String -> List},其中每个键是一个排序字符串,每个值是初始输入的字符串列表。
代码如下:
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
map<string,vector<string>> tmp;
vector<vector<string>> result;
vector<string> vec;
string str;
for(int i=0;i<strs.size();i++)
{
str = strs[i];
sort(str.begin(),str.end());
if(tmp.count(str))
tmp[str].push_back(strs[i]);
else
{
tmp[str] = vec;
tmp[str].push_back(strs[i]);
}
}
for(auto it=tmp.begin();it!=tmp.end();it++)
result.push_back(it->second);
return result;
}
};