问题描述: 输入n个整数,找出其中最小的k个数。
- 数据保证k一定小于等于输入数组的长度;
- 输出数组内元素请按从小到大顺序排序;
样例 - 输入:[1,2,3,4,5,6,7,8] , k=4
- 输出:[1,2,3,4]
好久不写c++ 代码了,介绍一个C++11上的新增功能auto关键字(使用auto可以简化for循环中的代码书写),在使用DevC++5.11编译运行时需要更改设置,否则无法通过编译在Tools–>Compiler options–>seetting–>code generation,如下图中设置编译环境,修改后即可。
应用大根堆实现此问题,c++中自带了大根堆C++ 标准库中的堆(heap):
- make_heap: 根据指定的迭代器区间以及一个可选的比较函数,来创建一个heap. O(N)
- push_heap: 把指定区间的最后一个元素插入到heap中. O(logN)
- pop_heap: 弹出heap顶元素, 将其放置于区间末尾. O(logN)
- sort_heap:堆排序算法,通常通过反复调用pop_heap来实现. N*O(logN)
C++11加入了两个新成员: - is_heap: 判断给定区间是否是一个heap. O(N)
- is_heap_until: 找出区间中第一个不满足heap条件的位置. O(N)
本次使用的是优先队列本质是也是使用堆实现的。链接: https://blog.csdn.net/weixin_36888577/article/details/79937886
class Solution {
public:
vector<int> getLeastNumbers_Solution(vector<int> input, int k) {
priority_queue<int> heap;
for(auto x:input )
{
heap.push(x);
if(heap.size()>k) heap.pop();
}
vector<int> res;
while(heap.size()) res.push_back(heap.top()),heap.pop();
reverse(res.begin(),res.end());
return res;
}
};
由于优先队列是大根堆,考虑到本文是做从小到大排序,并且只要K个数字。如果heap.size()>k,完全可以删除队头元素。剩下的K个元素即为要得到的元素,将元素依次读取队头,并保存在vector这一容器中,之后对该序列进行转置输出即为所求。
703. 数据流中的第 K 大元素
设计一个找到数据流中第 k 大元素的类(class)。注意是排序后的第 k 大元素,不是第 k 个不同的元素。请实现 KthLargest 类:
KthLargest(int k, int[] nums) 使用整数 k 和整数流 nums 初始化对象。
int add(int val) 将 val 插入数据流 nums 后,返回当前数据流中第 k 大的元素。
示例:
输入:
[“KthLargest”, “add”, “add”, “add”, “add”, “add”]
[[3, [4, 5, 8, 2]], [3], [5], [10], [9], [4]]
输出:
[null, 4, 5, 5, 8, 8]
解释:
KthLargest kthLargest = new KthLargest(3, [4, 5, 8, 2]);
kthLargest.add(3); // return 4
kthLargest.add(5); // return 5
kthLargest.add(10); // return 5
kthLargest.add(9); // return 8
kthLargest.add(4); // return 8
class KthLargest {
public:
priority_queue<int,vector<int>,greater<int>> q;
int knum;
KthLargest(int k, vector<int>& nums) {
knum=k;
for(auto x:nums) add(x);
}
int add(int val) {
q.push(val);
if(q.size()>knum) q.pop();
return q.top();
}
};