1. 程式人生 > 實用技巧 >leetcode347 前k個高頻元素(Medium)

leetcode347 前k個高頻元素(Medium)

題目來源:leetcode347 前k個高頻元素

題目描述:

給定一個非空的整數陣列,返回其中出現頻率前 k 高的元素。

示例 1:

輸入: nums = [1,1,1,2,2,3], k = 2
輸出: [1,2]

示例 2:

輸入: nums = [1], k = 1
輸出: [1]

提示:

你可以假設給定的 k 總是合理的,且 1 ≤ k ≤ 陣列中不相同的元素的個數。
你的演算法的時間複雜度必須優於 O(n log n) , n 是陣列的大小。
題目資料保證答案唯一,換句話說,陣列中前 k 個高頻元素的集合是唯一的。
你可以按任意順序返回答案。

解題思路:

用map儲存數字出現的次數,並進行排序。

class Solution {
public:
    static bool cmp(pair <int,int> p1,pair<int,int> p2){
        return p1.second>p2.second;
    }
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int,int> m;
        int i;
        for(i=0;i<nums.size();i++){
            m[nums[i]]++;
        }
        vector<int> ans;
        vector<pair<int,int>> res(m.begin(),m.end());
        sort(res.begin(),res.end(),cmp);
        for(i=0;i<k;i++){
            ans.push_back(res[i].first);
        }
        return ans;
    }
};

或者也可以用最小堆。