347. 前 K 个高频元素(STL)

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int> ans;
        vector<pair<int,int>> v;
        unordered_map<int,int> ma;
        for(int i=0;i<nums.size();i++){
            ma[nums[i]] ++;
        }
        for(auto i : ma){
            v.push_back(make_pair(i.first, i.second));
        }
        sort(v.begin(), v.end(), [](const pair<int,int> a,const pair<int,int> b){
            if(a.second == b.second)return a.first < b.first;
            return a.second > b.second;
        });
        for(int i=0;i<k;i++){
            ans.push_back(v[i].first);
        }
        return ans;
    }
};