给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]
说明:
- 所有输入均为小写字母。
- 不考虑答案输出的顺序。
这个题就是看每个词所包含的字母及其数量是否相等,如果相等就放在一起,所以可以用一个对应26个字母的数组来对字母计数,然后把这26个数字形成一个字符串,这样就形成了同字母数的一个对应,然后每个形成这个字符串的词都放在一起即可,这里用非排序map来存储,key就是字母数字符串,value就是一个类型的词的列表。
C++源代码:
class Solution {
public:
vector<vector<string>> groupAnagrams(vector<string>& strs) {
vector<vector<string>> res;
unordered_map<string, vector<string>> m;
for (string str : strs)
{
vector<int> count(26, 0);
string t = "";
for(char c : str)
count[c-'a']++;
for(int i : count)
t += to_string(i) + '-';
m[t].push_back(str);
}
for (auto s : m)
res.push_back(s.second);
return res;
}
};
python3源代码:
class Solution:
def groupAnagrams(self, strs):
"""
:type strs: List[str]
:rtype: List[List[str]]
"""
res = []
m = collections.defaultdict(list)
for s in strs:
count = [0 for i in range(26)]
t = ""
for c in s:
count[ord(c)-ord('a')] += 1
for i in count:
t += str(i) + '-'
m[t].append(s)
for t in m.values():
res.append(t)
return res
python这里用到了一个collection.defaultdict(list)的类,这个类在网上查是说可以没有键,它自己可以设置默认键。这里主要是用到了它的值可以是列表,然后这个列表添加所有的同类型词。