给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
输入: ["eat", "tea", "tan", "ate", "nat", "bat"]
,
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
说明:
- 所有输入均为小写字母。
- 不考虑答案输出的顺序。
比较简单的处理字符串的题,但我解的过程并不简单。思路就是对字符串做排序,然后存到对应的数组中。第一次做的想法是,遍历字符串数组,对每个字符串排序,再遍历数组,对每个数组中的第一个字符串进行排序,若两个字符串相等,则存到该数组中,思路和实现都很简单。代码如下:
class Solution {
public:
vector<vector<string> > groupAnagrams(vector<string>& strs)
{
vector<vector<string> > res;
for(int i=0;i<strs.size();i++)
{
int j;
string temp=strs[i];
sort(strs[i].begin(),strs[i].end());
for(j=0;j<res.size();j++)
{
string a=res[j][0];
sort(a.begin(),a.end());
if(strs[i]==a)
{
res[j].push_back(temp);
break;
}
}
if(j==res.size())
{
vector<string> a;
a.push_back(temp);
res.push_back(a);
}
}
return res;
}
};
但是超时了。。。。
于是就想着省去找对应数组的时间,就不需要对数组第一个字符串排序了。想到了用map,key就是排序后的字符串,value就是对应数组的索引。还是对字符串数组遍历并排序,再遍历map的key,若key中有这个排序后的字符串,则直接将原字符串存到索引为value的数组中;若遇到一个新的字符串,就把他放到一个新的数组中,同时更新map,map的key为排列后的字符串,value是新数组的索引。代码如下:
class Solution {
public:
vector<vector<string> > groupAnagrams(vector<string>& strs)
{
map<string,int> m;
map<string,int>::iterator it;//map的迭代器
vector<vector<string> > res;
for(int i=0;i<strs.size();i++)
{
string temp=strs[i];
sort(strs[i].begin(),strs[i].end());
for(it=m.begin();it!=m.end();it++)
{
if(it->first==strs[i])
{
res[it->second].push_back(temp);
break;
}
}
if(it==m.end())
{
vector<string> a;
a.push_back(temp);
res.push_back(a);
m[strs[i]]=res.size()-1;
}
}
return res;
}
};
没毛病,通过了,还学到了新的map遍历的方法,就是定义一个迭代器it,it->first就是key,it->second就是value。
但是一看时间,卧槽花了1000多ms,排在2%左右。。。于是乎看了排名第一的代码,差别不大,主要在于map的遍历上,人家用了map的find函数,find函数可以判断数据是否在map中,返回一个map中数据所在位置的迭代器。还有就是添加新的数组时,人家用了简便的方法,直接在push_back的括号里定义了一个vector,学到了学到了。代码如下:
class Solution {
public:
vector<vector<string> > groupAnagrams(vector<string>& strs)
{
map<string,int> m;
vector<vector<string> > res;
for(int i=0;i<strs.size();i++)
{
string temp=strs[i];
sort(strs[i].begin(),strs[i].end());
if(m.find(strs[i])!=m.end()) //find函数返回的是迭代器
res[m[strs[i]]].push_back(temp);
else
{
res.push_back(vector<string>(1,temp));//1表示有一个数据
m[strs[i]]=res.size()-1;
}
}
return res;
}
};