排序LeetCode 347 Top K Frequent Elements

LeetCode 347

Top K Frequent Elements

• Problem Description：
  返回数组出现频率最高的前k个元素（若从1开始计数，包含第k个）。
  具体的题目信息：
  https://leetcode.com/problems/top-k-frequent-elements/description/
• Example:
  Given [1,1,1,2,2,3] and k = 2, return [1,2].
• Solution:
  
  • 解题思路：先用Map容器存储每个元素及其出现的次数，然后转而用vector容器存储方便使用自定义的比较函数（按照出现次数从大到小依次排列），最后依次输出前k个元素。
  • 编程实现：

class Solution {
public:
    static bool cmp(pair<int, int>& p, pair<int, int>&q) {
        return p.second>q.second;
    }
    vector<int> topKFrequent(vector<int>& nums, int k) {
        if (nums.size() == 0||nums.size() == 1) return nums;
        map<int, int> res;
        for (int i = 0; i < nums.size(); i++) {
            res[nums[i]]++;
        }
        vector<pair<int,int>> v(res.begin(), res.end());
        sort(v.begin(), v.end(), cmp);
        vector<int> arr;
        for (int i = 0; i < k; i++) {
            arr.push_back(v[i].first);
        }
        return arr;
    }
};