LeetCode 347. Top K Frequent Elements

解题思路:这一题和leetcode451题的思路是类似的,只不过是返回前topK出现最多的频率,所以自然而然地相当先将vector写进map中,然后根据value值排序,注意题目要求的时间复杂度,使用全排sort虽然也是采用的快排思想,有时也会达到O(N2),所以使用部分排序,堆排序,C++ stl里面有partial_sort,仅对前K个排序.

class Solution {public:    vector<int> topKFrequent(vector<int>& nums, int k) {        map<int,int>tmp;        vector<int>k_nums;        for(int i = 0; i < nums.size(); i++){            tmp[nums[i]]++;        }        vector<pair<int, int> > res(tmp.begin(),tmp.end());        partial_sort(res.begin(), res.begin()+k, res.end(), CmpByValue());        for(int i = 0;i < k; i++){          k_nums.push_back(res[i].first);          }        return k_nums;    }private:    struct CmpByValue {      bool operator()(const pair<int,int>& p1, const pair<int,int>& p2) {      return p1.second > p2.second;    }  };  };