【牛客】最小的k个数
来源:互联网 发布:民科笑话知乎 编辑:程序博客网 时间:2024/05/16 14:03
使用topK的方法,如果不想写堆的话,了解几个库里的堆的函数
std::make_heap将[start, end)范围进行堆排序,默认使用less, 即最大元素放在第一个。
std::pop_heap将front(即第一个最大元素)移动到end的前部,同时将剩下的元素重新构造成(堆排序)一个新的heap。
std::push_heap对刚插入的(尾部)元素做堆排序。
std::sort_heap将一个堆做排序,最终成为一个有序的系列,可以看到sort_heap时,必须先是一个堆(两个特性:1、最大元素在第一个 2、添加或者删除元素以对数时间),因此必须先做一次make_heap.
思路:
1、参数检验:len<=0、k>len、k<=0 都是不对的
2、建堆:使用make_heap建立k个节点的最大堆
3、使用剩余元素和堆顶元素比较,
4、如果小于堆顶元素,那么删除堆顶元素,将此元素插入
5、最后使用sort_heap排序,输出
class Solution {public: vector<int> GetLeastNumbers_Solution(vector<int> input, int k) { //topK int len=input.size(); if(len<=0||k>len||k<=0) return vector<int>(); //建立k个元素的堆(大堆) vector<int> res; for(int i=0;i<k;i++){ res.push_back(input[i]); } make_heap(res.begin(),res.end()); for(int i=k;i<len;i++){ if(res[0]>input[i]){ pop_heap(res.begin(),res.end());//将待删除元素放到最后 res.pop_back(); res.push_back(input[i]); push_heap(res.begin(),res.end()); } } //最后,再从小到大输出 sort_heap(res.begin(),res.end()); return res; }};
阅读全文
0 0
- 【牛客】最小的k个数
- 最小的K个数
- 最小的K个数
- 最小的K个数
- 最小的k个数
- 最小的k个数
- 最小的k个数
- 最小的k个数
- 最小的k个数
- 最小的k个数
- 最小的K个数
- 最小的k个数
- 最小的K个数
- 最小的K个数
- 最小的K个数
- 最小的K个数
- 最小的K个数
- 最小的k个数
- linux 网络命令
- 蓝牙核心系统架构介绍
- linux 命令行中的计算器
- 数据库视频总结 一
- [Android6.0][RK3399] TypeC 基础知识
- 【牛客】最小的k个数
- 【转载】SSM框架——详细整合教程(Spring+SpringMVC+MyBatis)
- Eclipse中安装SVN插件
- 位操作
- bash配置文件/etc/profile,/etc/bashrc,~/.bash_profile,~/.bashrc详解及修改PATH环境变量
- 50 种系统免遭黑客侵袭的方法 [2017 年版]
- jsp中page的各个属性介绍
- android改进布局性能
- Android中Vitamio全屏播放