[CareerCup] 11.2 Sort Anagrams Array 異位詞陣列排序

 

11.2 Write a method to sort an array of strings so that all the anagrams are next to each other.

 

這道題讓我們給一個字串陣列排序，讓所有的變位詞Anagrams排在一起，關於變位詞，LeetCode裡有兩道相關的題目Anagrams 錯位詞和Valid Anagram 驗證變位詞。那麼對於這道題，我們有兩種方法可以實現，先來看第一種方法，來重寫sort中的比較函式compare，參見程式碼如下：

 

解法一：

bool cmp(const string &a, const string &b) {
    string m = a, n = b;
    sort(m.begin(), m.end());
    sort(n.begin(), n.end());
    return !m.compare(n);
}

sort(array.begin(), array.end(), cmp);

 

另一種解法較為複雜一些，用到了雜湊表來建立排序後的字串和其所有的異位詞集合的對映，最後在按集合填充原陣列，參見程式碼如下：

 

解法二：

class Solution {
public:
    void sortArray(vector<string> &array) {
        unordered_map<string, vector<string> > m;
        for (auto &a : array) {
            string key = a;
            sort(key.begin(), key.end());
            if (m.find(key) == m.end()) {
                m[key] = vector<string>();
            }
            vector<string> &v = m[key];
            v.push_back(a);
        }
        int idx = 0;
        for (unordered_map<string, vector<string> >::iterator it = m.begin(); it != m.end(); ++it) {
            for (auto &a : it->second) {
                array[idx++] = a;
            }
        }
    }
};