题目链接

https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?tpId=13&tqId=11182&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

题意

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的四个数字是1,2,3,4,

解题思路

法一:若可以修改输入数据:时间复杂度O(n)。

利用快排的partition数组。第一遍以后partion的范围要想清楚,"是一个小小的范围",这样才能实现时间复杂度O(n)。

注意:1、特判一定要全!2、瞎throw容易看不到真正的错误提示... 3、各种边界填写的变量要过脑子...

法二:若不可以修改输入数据,或是要处理海量数据(内存较小):时间复杂度O(nlogk),内存要求可容纳大小为k的数组即可。

  • 可以维护一个大小为k的数组。
  • 容器满了后,每次再来一个数,若大于容器内最大值,则删除、插入新值(即替换),若小于容器内最大值,则忽略。
  • 最终得到的数组k即是最小的k个

使用红黑树实现以上算法。

特别地,由于直接替换节点可能破坏红黑树的结构,所以红黑树不支持直接改变元素值,必须要删除旧值、插入新值来实现。

相关知识

红黑树
  • 红黑树是特殊的二叉查找树,红黑节点保证最长路径<=2*最短路径。红黑树是自动有序的。

  • 红黑树不严格满足二叉平衡树(AVL)左右子树高度最多相差1的条件。?具体

  • 红黑树的查找、插入、删除时间复杂度都是O(logk)。?具体

  • STL中的set和multiset都是用红黑树实现,其中multiset可有重复元素。

比较仿函数greater 与less (即可以当作函数一样使用)
  • typedef multiset<int,greater<int>> intSet中,greater实质是一个结构体,greater表示内置类型从大到小排序,less表示内置类型从小到大排序。它们也可以用作sort函数的比较器。
  • set与multiset中默认使用less,即由小到大排序。
最大堆

最大堆中,根结点的值总是大于它的子树中任意节点的值。

得到最大值O(1)、插入、删除时间复杂度都是O(logk)。

代码

法一代码

class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> minNums;
if(input.size()!=0&&k>=1&&k<=input.size()){//!!!写全
size_t beg=0;
size_t end=input.size()-1;
size_t index=partition(input,beg,end);//
while(index!=k-1){//
if(index<k-1){
beg=index+1;//
index=partition(input,beg,end);//
}
else{
end=index-1;//
index=partition(input,beg,end);//
}
}
for(int i=0;i<k;i++){
minNums.push_back(input[i]);
}
}
return minNums;
}
private:
size_t partition(vector<int> &input,size_t beg,size_t end){
if(!input.size()||end<beg||beg<0){
throw "input error!";
}
size_t index=random(beg,end);
swap(input[index],input[beg]);//
int temp=input[beg]; size_t i=beg;
size_t j=end;
while(i<j){
while(i<j&&input[j]>=temp){
--j;
}
if(i!=j){
input[i]=input[j];
++i;
}
while(i<j&&input[i]<=temp){
++i;
}
if(i!=j){
input[j]=input[i];
--j;
}
}
input[i]=temp;
return i;
} void swap(int& a,int& b){
int temp=a;
a=b;
b=temp;
} size_t random(size_t beg,size_t end){
if(end<beg){
throw "input error!";
}
srand((unsigned) time(0));
return beg+rand()%(end-beg+1);
}
};

法二代码

class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> minVec;
typedef multiset<int,greater<int>> intSet;
intSet minSet;
if(input.size()&&k>0&&k<=input.size()){
for(auto it=input.begin();it!=input.end();++it){
if(minSet.size()!=k){
minSet.insert(*it);
}
else{
if(*minSet.begin()>*it){
minSet.erase(minSet.begin());
minSet.insert(*it);
}
}
}
}
for(auto it=minSet.begin();it!=minSet.end();++it){
minVec.push_back(*it);
}
return minVec;
}
};

[剑指Offer]40-最小的k个数的更多相关文章

  1. 剑指 Offer 40. 最小的k个数 + 优先队列 + 堆 + 快速排序

    剑指 Offer 40. 最小的k个数 Offer_40 题目描述 解法一:排序后取前k个数 /** * 题目描述:输入整数数组 arr ,找出其中最小的 k 个数.例如,输入4.5.1.6.2.7. ...

  2. 剑指 Offer 40. 最小的k个数

    剑指 Offer 40. 最小的k个数 输入整数数组 arr ,找出其中最小的 k 个数.例如,输入4.5.1.6.2.7.3.8这8个数字,则最小的4个数字是1.2.3.4. 示例 1: 输入:ar ...

  3. 【Java】 剑指offer(40) 最小的k个数

    本文参考自<剑指offer>一书,代码采用Java语言. 更多:<剑指Offer>Java实现合集   题目 输入n个整数,找出其中最小的k个数.例如输入4.5.1.6.2.7 ...

  4. 每日一题 - 剑指 Offer 40. 最小的k个数

    题目信息 时间: 2019-06-30 题目链接:Leetcode tag: 快排 难易程度:中等 题目描述: 输入整数数组 arr ,找出其中最小的 k 个数.例如,输入4.5.1.6.2.7.3. ...

  5. 【剑指Offer】最小的K个数 解题报告(Python)

    [剑指Offer]最小的K个数 解题报告(Python) 标签(空格分隔): 剑指Offer 题目地址:https://www.nowcoder.com/ta/coding-interviews 题目 ...

  6. Go语言实现:【剑指offer】最小的K个数

    该题目来源于牛客网<剑指offer>专题. 输入n个整数,找出其中最小的K个数.例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,. Go语言实现: fu ...

  7. 剑指OFFER之最小的K个数(九度OJ1371)

    题目描述: 输入n个整数,找出其中最小的K个数.例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,. 输入: 每个测试案例包括2行: 第一行为2个整数n,k(1< ...

  8. 《剑指offer》最小的k个数

    本题来自<剑指offer> 反转链表 题目: 思路: C++ Code: Python Code: 总结:

  9. 剑指Offer 29. 最小的K个数 (其他)

    题目描述 输入n个整数,找出其中最小的K个数.例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4. 题目地址 https://www.nowcoder.com/prac ...

  10. 【剑指offer】最小的K个数

    一.题目: 输入n个整数,找出其中最小的K个数.例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,. 二.思路: 一群大牛在讨论用噼里啪啦各种排序,复杂度一般也都是O ...

随机推荐

  1. 深度学习原理与框架-神经网络-cifar10分类(代码) 1.np.concatenate(进行数据串接) 2.np.hstack(将数据横着排列) 3.hasattr(判断.py文件的函数是否存在) 4.reshape(维度重构) 5.tanspose(维度位置变化) 6.pickle.load(f文件读入) 7.np.argmax(获得最大值索引) 8.np.maximum(阈值比较)

    横1. np.concatenate(list, axis=0) 将数据进行串接,这里主要是可以将列表进行x轴获得y轴的串接 参数说明:list表示需要串接的列表,axis=0,表示从上到下进行串接 ...

  2. day06-单表查询

    1.单表查询的语法 SELECT 字段1,字段2... FROM 表名 WHERE 条件 GROUP BY field HAVING 筛选 ORDER BY field LIMIT 限制条数 2.关键 ...

  3. jsfl 将库中声音放置到时间轴上

    setBrokenFrames("cz1.mp3",2); /**/ /*<--------------将库中声音放置到时间轴上------------->*/ /** ...

  4. C++复习:继承与派生

    1继承概念 面向对象程序设计有4个主要特点:抽象.封装.继承和多态性.说了类和对象,了解了面向对象程序设计的两个重要特征一数据抽象与封装,已经能够设计出基于对象的程序,这是面向对象程序设计的基础. 要 ...

  5. Object-c @property与@synthesize的配对使用。

    功能:让编译器自动编写一个与数据成员同名的方法声明来省去读写方法的声明. 如: 1.在头文件中: @property int count; 等效于在头文件中声明2个方法: - (int)count; ...

  6. maven ,添加加密算法,使用

    1:消息摘要:(数字指纹):既对一个任意长度的一个数据块进行计算,产生一个唯一指纹.MD5/SHA1发送给其他人你的信息和摘要,其他人用相同的加密方法得到摘要,最后进行比较摘要是否相同. MD5(Me ...

  7. 【原创】利用Office宏实现powershell payload远控

    本文将演示使用Veil-Evasion生成远控所需要的payload以及监听器,然后使用MacroShop生成payload 相关的VBA代码,最后演示将VBA代码写入.doc文本文档的宏中. 环境: ...

  8. python函数基础:调用内置函数&定义函数

    调用内置函数 有很多内置函数,在使用中需要积累.这里只举两个例子: 分别调用abs和数据类型转换,注意当入参类型错误时候会报错 ''' print('abs(-100)') abs(-100) pri ...

  9. javascript原型链中 this 的指向

    为了弄清楚Javascript原型链中的this指向问题,我写了个代码来测试: var d = { d: 40 }; var a = { x: 10, calculate: function (z) ...

  10. C# 递归获取 文件夹的 所有文件

    public void Director(string dir, List<string> list) { DirectoryInfo d = new DirectoryInfo(dir) ...