您的位置:首页 > 产品设计 > UI/UE

【leetcode】347. Top K Frequent Elements

2016-07-24 21:09 399 查看
一、题目描述

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 可以接受隐式的类型转换,这样可以获得更高的灵活度
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: