leetcode-347. 前 K 个高频元素

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());//先排序
        unordered_map<int, int> Map;//记录不同元素出现的次数
        vector<int> com1;//记录所有出现过得不同的元素
        vector<int> com2;//记录不同元素出现的次数
        vector<int> ans;//返回值
        for (int i = 0; i < (int)nums.size(); i++)//用hashmap统计元素出现次数，用com1记录元素
        {
            Map[nums[i]]++;
            if (com1.size() == 0 || com1.back() != nums[i])
                com1.push_back(nums[i]);
        }
        for (int i = 0; i < (int)com1.size(); i++)//将元素出现次数存进com2方便找最大值
        {
            com2.push_back(Map[com1[i]]);
        }
        for (int i = 0; i < k; i++)
        {
            int _max = max_element(com2.begin(), com2.end())-com2.begin();//返回当前所有数中出现最多次元素的下标
            ans.push_back(com1[_max]);
            com2[_max] = -1;//去掉已经记录在内的元素
        }
        return ans;
    }
};