1. 程式人生 > >LeetCode演算法題49:字母異位詞分組解析

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)的類,這個類在網上查是說可以沒有鍵,它自己可以設定預設鍵。這裡主要是用到了它的值可以是列表,然後這個列表新增所有的同類型詞。