题目要求
Given an array of strings, group anagrams together.
For example, given: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”],
Return:
[
[“ate”, “eat”,”tea”],
[“nat”,”tan”],
[“bat”]
]
Note: All inputs will be in lower-case.
给定一个字符串的数组,将其中的同构体(类似于“eat”和“tea”)找出来放在一组。输出结果保存这样一个分组。
解题思路1
时间复杂度O(n2):
对于每一个字符串,创建一个统计它字母的vector<\int> v(26),如果两个字符串的v统计数组相同,则为同构体。
用一个map来记录v统计数组和字符串之间的联系:
map<vector<int>,vector<int>> m;
//m.first为统计数组,m.second表示满足这个统计数组的字符串在数组中的下标集合。
则先遍历字符串数组,计算出v统计数组,然后检查在map中是否有出现同样的统计数组,如果有,则在该map元素的second中加入该字符串的下标;如果无,则直接加入map中。
再对map进行遍历,将每一个map元素中的second下标值的字符串找出放在一组中。
C++代码
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
map<vector<int>, vector<int>> m;
for (int i = 0; i<strs.size(); i++) {
vector<int> v = cal(strs[i]);
map<vector<int>, vector<int>>::iterator it;
for (it = m.begin(); it != m.end(); it++) {
if (it->first == v) {
it->second.push_back(i);
break;
}
}
if (it == m.end()) {
vector<int> tmp;
tmp.push_back(i);
m[v] = tmp;
}
}
vector<vector<string>> res;
for (auto it = m.begin(); it != m.end(); it++) {
vector<string> tmp;
for (int i = 0; i<it->second.size(); i++) {
tmp.push_back(strs[it->second[i]]);
}
res.push_back(tmp);
}
return res;
}
vector<int> cal(string a) {
vector<int> v(26, 0);
for (int i = 0; i<a.length(); i++) {
v[a[i] - 'a']++;
}
return v;
}
};
最后结果
101 / 101 test cases passed.
Status: Accepted
Runtime: 1731 ms
差点超时
解题思路2
还是利用map,只是map的第一个元素是排序字符串,第二个元素里装有排好序是第一个元素的字符串数组。首先遍历输入数组,将该map构造好,再将该map遍历,就得到了分好组的数组。
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
map<string,multiset<string>> m;
for(int i=0;i<strs.size();i++){
string tmp = strs[i];
sort(tmp.begin(),tmp.end());
m[tmp].insert(strs[i]);
}
vector<vector<string>> res;
for (auto it=m.begin();it!=m.end();it++){
vector<string> v(it->second.begin(),it->second.end());
res.push_back(v);
}
return res;
}
};
最后结果
101 / 101 test cases passed.
Status: Accepted
Runtime: 39 ms