LeetCode演算法題49:字母異位詞分組解析
阿新 • • 發佈:2019-01-03
給定一個字串陣列,將字母異位詞組合在一起。字母異位詞指字母相同,但排列不同的字串。
示例:
輸入: ["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)的類,這個類在網上查是說可以沒有鍵,它自己可以設定預設鍵。這裡主要是用到了它的值可以是列表,然後這個列表新增所有的同類型詞。