[LeetCode] 49. Group Anagrams 群組錯位詞

Given an array of strings, group anagrams together.

Example:

Input: ["eat", "tea", "tan", "ate", "nat", "bat"],
Output:
[
["ate","eat","tea"],
["nat","tan"],
["bat"]
]

Note:

• All inputs will be in lowercase.
• The order of your output does not matter.

這道題讓我們?nèi)航M給定字符串集中所有的錯位詞，所謂的錯位詞就是兩個字符串中字母出現(xiàn)的次數(shù)都一樣，只是位置不同，比如 abc，bac, cba 等它們就互為錯位詞，那么如何判斷兩者是否是錯位詞呢，可以發(fā)現(xiàn)如果把錯位詞的字符順序重新排列，那么會得到相同的結(jié)果，所以重新排序是判斷是否互為錯位詞的方法，由于錯位詞重新排序后都會得到相同的字符串，以此作為 key，將所有錯位詞都保存到字符串數(shù)組中，建立 key 和當前的不同的錯位詞集合個數(shù)之間的映射，這里之所以沒有建立 key 和其隸屬的錯位詞集合之間的映射，是用了一個小 trick，從而避免了最后再將 HashMap 中的集合拷貝到結(jié)果 res 中。當檢測到當前的單詞不在 HashMap 中，此時知道這個單詞將屬于一個新的錯位詞集合，所以將其映射為當前的錯位詞集合的個數(shù)，然后在 res 中新增一個空集合，這樣就可以通過其映射值，直接找到新的錯位詞集合的位置，從而將新的單詞存入結(jié)果 res 中，參見代碼如下：

解法一：

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>> res;
        unordered_map<string, int> m;
        for (string str : strs) {
            string t = str;
            sort(t.begin(), t.end());
            if (!m.count(t)) {
                m[t] = res.size();
                res.push_back({});
            }
            res[m[t]].push_back(str);
        }
        return res;
    }
};

下面這種解法沒有用到排序，用一個大小為 26 的 int 數(shù)組來統(tǒng)計每個單詞中字符出現(xiàn)的次數(shù)，然后將 int 數(shù)組轉(zhuǎn)為一個唯一的字符串，跟字符串數(shù)組進行映射，這樣就不用給字符串排序了，參見代碼如下：

解法二：

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> cnt(26);
            string t;
            for (char c : str) ++cnt[c - 'a'];
            for (int i = 0; i < 26; ++i) {
                if (cnt[i] == 0) continue;
                t += string(1, i + 'a') + to_string(cnt[i]);
            }
            m[t].push_back(str);
        }
        for (auto a : m) {
            res.push_back(a.second);
        }
        return res;
    }
};

到此這篇關(guān)于C++實現(xiàn)LeetCode(49.群組錯位詞)的文章就介紹到這了,更多相關(guān)C++實現(xiàn)群組錯位詞內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論