1. 程式人生 > >[CareerCup] 11.2 Sort Anagrams Array 異位詞陣列排序

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