leetcode 347 map根据value排序

Given a non-empty array of integers, return the k most frequent elements.

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note: 

• You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
• Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

Subscribe to see which companies asked this question.

思路：

1.遍历数组，map<key, num>插入，key为数字，num为出现次数

2.使用vector<pair<int,int>>赋值，然后在根据second排序~

代码：

class Solution {
public:
    static bool cmp(pair<int, int> a, pair<int, int> b)
    {
        return a.second > b.second;
    }
    vector<int> topKFrequent(vector<int>& nums, int k) {
        int size = nums.size();
        if(size == 0 || k > size)
            return nums;
        map<int, int> my_map;
        map<int, int>::iterator it;
        for (int i = 0 ; i < nums.size() ; i ++)
        {
            it = my_map.find(nums[i]);
            if(it == my_map.end())
                my_map[nums[i]] = 1;
            else
                my_map[nums[i]] += 1;
        }
        vector<pair<int, int>> a;
        for (it = my_map.begin() ; it != my_map.end() ; it ++)
        {
            a.push_back(pair<int, int>(it->first, it->second));
        }
        sort(a.begin(), a.end(), cmp);
        vector <int> res;
        for (int i = 0 ; i < k ; i++)
            res.push_back(a[i].first);
        return res;
    }
};