LeetCode Anagrams

struct mystat {
    int idx;
    int cnt[26];
    mystat(int id = 0) {idx = id;}
};

bool cmp(const mystat &a, const mystat &b) {
    for (int i=0; i<26; i++) {
        if (a.cnt[i] < b.cnt[i]) {
            return true;
        } else if (a.cnt[i] > b.cnt[i]) {
            break;
        } else {
            continue;
        }
    }
    return false;
}

bool same(const mystat &a, const mystat &b) {
    for (int i=0; i<26; i++) {
        if (a.cnt[i] != b.cnt[i]) return false;
    }
    return true;
}

class Solution {
public:
    vector<string> anagrams(vector<string> &strs) {
        vector<string> res;
        vector<mystat> stats;

        int len = strs.size();

        for (int i=0; i<len; i++) {
            stats.push_back(mystat(i));

            for (int j=strs[i].size() - 1; j>=0; j--) {
                stats.back().cnt[strs[i][j] - 'a']++;
            }
        }
        
        sort(stats.begin(), stats.end(), cmp);

        int si = 0;

        while (si < len ) {
            int i = si + 1;
            for (; i<len; i++) {
                if (same(stats[si], stats[i])) {
                    res.push_back(strs[stats[i].idx]);
                } else {
                    break;
                }
            }
            if (si + 1 < i) {
                res.push_back(strs[stats[si].idx]);
            }
            si = i;
        }
        return res;
    }
    
};

虽然做出来了，不过写那么长，时间又是150ms+肯定还有什么巧妙的方法。

简短的版本：

class Solution {
public:
    vector<string> anagrams(vector<string>& strs) {
        unordered_map<string, int> count;
        vector<string> res;
        int len = strs.size();
        for (int i=0; i<len; i++) {
            string str = strs[i];
            sort(str.begin(), str.end());
            if (count.count(str) == 0) {
                count[str] = i;
            } else {
                if (count[str] >= 0) {
                    res.push_back(strs[count[str]]);
                    count[str] = -1;
                }
                res.push_back(strs[i]);
            }
        }
        return res;
    }
};