一、题目描述
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.
题目解读:给一个整数数组nums,和一个数k,输出nums中出现频次最多的前k个数的数组
思路:
1、求出nums中每个数出现的频次
2、按频次进行排序
3、输出前k个数组成的数组
本题思路不难,难就难在需要思考一下用什么数据结构去实现比较简单,比较方便,并且复杂度低。
c++代码(36ms,75.46%)
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
vector<int> result;
//1、使用map存储每个interger和出现的次数,统计词频
unordered_map<int,int> count;
for(int number : nums)
count[number]++;
//2、根据频次压入最大堆中
priority_queue<pair<int, int>> p; //最大堆
for(auto it = count.begin(); it != count.end(); it++){
p.push(make_pair(it->second, it->first));
}//for
//3、获取结果
while(result.size() < k){
result.push_back(p.top().second);
p.pop();
}//while
return result;
}
};
总结:
(1)map 与 unordered_map比较
map
优点:有序性,
红黑树,内部实现一个红黑树使得 map 的很多操作在logn的时间复杂度下就可以实现,因此效率非常的高
缺点:空间占用率高,因为 map 内部实现了红黑树,虽然提高了运行效率,但是因为每一个节点都需要额外保存父节点,孩子节点以及红/黑性质,使得每一个节点都占用大量的空间
适用处:对于那些有顺序要求的问题,用 map 会更高效一些
unordered_map
优点:因为内部实现了哈希表,因此查找速度非常的快
缺点:哈希表的建立比较耗费时间
适用处:对于查找问题,unordered_map 会更加高效一些,因此遇到查找问题,常会考虑一下用 unordered_map
(2)c++ std:: pair 与 std:: make_pair
pair主要的作用是将两个数据组合成一个数据,两个数据可以是同一类型或者不同类型。例如pair<int, float> 或者 pair<double, double>等。pair 实质上是一个结构体,其主要的两个成员变量是 first 和 second, 这两个变量可以直接使用。
make_pair一般都使用在需要 pair 做参数的位置,可以直接调用 make_pair 生成 pair对象,另一个使用的方面是pair 可以接受隐式的类型转换,这样可以获得更高的灵活度