C++實現LeetCode(49.群組錯位詞)

Posted on by

[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.

這道題讓我們群組給定字符串集中所有的錯位詞，所謂的錯位詞就是兩個字符串中字母出現的次數都一樣，隻是位置不同，比如 abc，bac, cba 等它們就互為錯位詞，那麼如何判斷兩者是否是錯位詞呢，可以發現如果把錯位詞的字符順序重新排列，那麼會得到相同的結果，所以重新排序是判斷是否互為錯位詞的方法，由於錯位詞重新排序後都會得到相同的字符串，以此作為 key，將所有錯位詞都保存到字符串數組中，建立 key 和當前的不同的錯位詞集合個數之間的映射，這裡之所以沒有建立 key 和其隸屬的錯位詞集合之間的映射，是用瞭一個小 trick，從而避免瞭最後再將 HashMap 中的集合拷貝到結果 res 中。當檢測到當前的單詞不在 HashMap 中，此時知道這個單詞將屬於一個新的錯位詞集合，所以將其映射為當前的錯位詞集合的個數，然後在 res 中新增一個空集合，這樣就可以通過其映射值，直接找到新的錯位詞集合的位置，從而將新的單詞存入結果 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 數組來統計每個單詞中字符出現的次數，然後將 int 數組轉為一個唯一的字符串，跟字符串數組進行映射，這樣就不用給字符串排序瞭，參見代碼如下：

解法二：

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;
    }
};

到此這篇關於C++實現LeetCode(49.群組錯位詞)的文章就介紹到這瞭,更多相關C++實現群組錯位詞內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：