LeetCode算法题49：字母异位词分组解析

给定一个字符串数组，将字母异位词组合在一起。字母异位词指字母相同，但排列不同的字符串。

示例:

输入: ["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)的类，这个类在网上查是说可以没有键，它自己可以设置默认键。这里主要是用到了它的值可以是列表，然后这个列表添加所有的同类型词。

LeetCode算法题49：字母异位词分组解析

猜你喜欢