排序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;
    }
};