LeetCode-791-自定义字符串排序

1、哈希表

我们可以利用哈希表记录每一位字符对应的数字，若字符在order中出现则对应其出现的位数，否则对应其ASCII码值。最终我们可以将字符串转化为一个数组而后对数组进行排序，最后将数组转化为字符串即可。

class Solution {
public:
    string customSortString(string order, string s) {
        unordered_map<char, int> hs;
        vector<int> temp;
        string res;
        for (int i = 0; i < order.size(); ++i) {
            hs[order[i]] = i;
        }
        for (int i = 0; i < s.size(); ++i) {
            if (hs.count(s[i])) temp.emplace_back(hs[s[i]]);
            else temp.emplace_back((int) s[i]);
        }
        sort(temp.begin(), temp.end());
        for (int i: temp) {
            if (i >= 'a') {
                res.push_back((char) i);
            } else {
                for (auto j: hs) {
                    if (j.second == (char) i) {
                        res.push_back(j.first);
                    }
                }
            }
        }
        return res;
    }
};

2、排序优化

我们可以自定义cmp函数直接对字符串进行排序。

class Solution {
public:
    string customSortString(string order, string s) {
        vector<int> val(26);
        for (int i = 0; i < order.size(); ++i) {
            val[order[i] - 'a'] = i + 1;
        }
        sort(s.begin(), s.end(), [&](char c0, char c1) {
            return val[c0 - 'a'] < val[c1 - 'a'];
        });
        return s;
    }
};

3、计数排序

考虑到出现的字符个数优先，我们可以使用计数排序对排序过程进行优化。我们可以利用哈希表记录s中每个字符出现的次数，而后遍历order根据哈希表中的数据直接在答案结尾进行添加即可。

class Solution {
public:
    string customSortString(string order, string s) {
        vector<int> freq(26);
        for (char ch: s) {
            ++freq[ch - 'a'];
        }
        string ans;
        for (char ch: order) {
            if (freq[ch - 'a'] > 0) {
                ans += string(freq[ch - 'a'], ch);
                freq[ch - 'a'] = 0;
            }
        }
        for (int i = 0; i < 26; ++i) {
            if (freq[i] > 0) {
                ans += string(freq[i], i + 'a');
            }
        }
        return ans;
    }
};