Leetcode 6121 日志

class Solution {
public:
    static bool mycomp(pair<string, int> a, pair<string, int> b) {
        return a.first < b.first;
    }
    vector<int> smallestTrimmedNumbers(vector<string>& nums, vector<vector<int>>& queries) {
        vector<int> ret;
        unordered_map<int, vector<pair<string, int>>> query_map;
        for (vector<int> &query : queries) {
            if (!query_map.count(query[1])) {
                vector<pair<string, int>> subnum;
                for (int i = 0; i < nums.size(); ++i) {
                    string &s = nums[i];
                    subnum.push_back({s.substr(s.size() - query[1]), i});
                }
                stable_sort(subnum.begin(), subnum.end(), mycomp);
                query_map[query[1]] = subnum;
            }

            ret.push_back(query_map[query[1]][query[0]-1].second);
        }
        return ret;
    }
};

std::sort 使用的快速排序是不稳定的，导致在测试用令时有时通过有时不通过，需要改用 std::stable_sort。

标签：sort,map,subnum,ret,vector,6121,query,日志,Leetcode
来源： https://www.cnblogs.com/3yude/p/16486916.html