【刷题之路】最小的K个数

原创 2016年07月29日 17:59:48

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

最优方法可以做到n*log(k)。做法就是使用堆排序,只维护一个规模为k的大根堆,之后遍历所有的数字,一旦后面数字比堆顶小,则交换两数,并重新调整大根堆,遍历完成后堆中元素就是最小的k个数(但是这K个数不一定是排序的)

代码如下:

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k){
        int i,j=0;
        vector<int> res;
        int n=input.size();
        if(n<k) return res; //如果K比总数还大,直接返回
        maxheap(input,k); //由前K个数建立一个大根堆
        for(i=k;i<n;i++){
            if(input[0]>input[i]) swap(input[0],input[i]);  //之后的数字只要比堆顶大,则改变堆顶,并调整K规模的大根堆
            fixdown(input,0,k);
        }
        for(i=0;i<k;i++) res.push_back(input[i]); //遍历完成后,输出前K个即为最小的K个数
        return res;
        
    }
    void maxheap(vector<int> &num,int n){
        int i;
        for(i=n/2-1;i>=0;i--){
            fixdown(num,i,n-1);
        }
    }
    void fixdown(vector<int> &num, int i,int n){ //这里的n是下标,所以之前的fixdown要是n-1
        int j;
        int flag=1;
        while(2*i+1<n && flag){  //常规堆排序做法,不详述
            if(2*i+2>=n) j=2*i+1;
            else{
                if(num[2*i+2]<num[2*i+1]) j=2*i+1;
                else j=2*i+2;
            }
            if(num[i]<num[j]){
                swap(num[i],num[j]);
                i=j;
            }
            else flag=0;
        }
    }
};

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zyn2609530/article/details/52067448

编程算法 - 最小的k个数 代码(C)

最小的k个数 代码(C)本文地址: http://blog.csdn.net/caroline_wendy题目: 输入n个整数, 找出其中的最小k个数.使用快速排序(Quick Sort)的方法求解,...
  • u012515223
  • u012515223
  • 2014-06-29 21:32:05
  • 1709

牛客网刷题之最小的K个数

题目描述:解题思路:  首先向导的是快排,然后取前k个数就可以了:题解: public ArrayList GetLeastNumbers_Solution(int[] input, int k) {...
  • lks1139230294
  • lks1139230294
  • 2016-10-18 11:04:49
  • 201

剑指offer-面试题30:最小的K个数

题目:输入n个数,找出其中最小的K个数。例如输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1,2,3,4。 思路:这道题目是典型的top K问题。两种方法:(1)如果允许改变数列,半...
  • moses1213
  • moses1213
  • 2016-04-06 21:10:02
  • 778

【剑指Offer面试题】 九度OJ1371:最小的K个数

题目链接地址: http://ac.jobdu.com/problem.php?pid=1371题目1371:最小的K个数时间限制:1 秒内存限制:32 兆特殊判题:否提交:5938解决:1265...
  • zhoubin1992
  • zhoubin1992
  • 2015-08-07 16:53:08
  • 765

剑指Offer:面试题30——最小的k个数(java实现)

问题描述: 输入n个整数,找出其中最小的k个数 思路1: 先排序,再取前k个 时间复杂度O(nlogn) 下面给出快排序的代码(基于下面Partition函数的方法)public voi...
  • lilianforever
  • lilianforever
  • 2016-07-08 16:05:00
  • 1046

输入n个整数,找出其中最小的K个数。——冒泡排序实现方法

import java.util.ArrayList; public class Solution {     /*          */     public ArrayList Get...
  • chainiao_zhang
  • chainiao_zhang
  • 2017-04-20 20:54:43
  • 229

[剑指Offer]最小的K个数

题目描述 输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。 输入 int[] input:待搜索的数组 int ...
  • shakespeare001
  • shakespeare001
  • 2016-04-29 14:29:06
  • 3673

剑指Offer面试题30(java版):最小的k个数

题目:输入n个整数,找出其中最小的k个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4...
  • jsqfengbao
  • jsqfengbao
  • 2015-08-08 20:47:08
  • 3208

最小的k个数(Java实现)

本题为剑指offer面试题30 牛客网测试地址:https://www.nowcoder.com/questionTerminal/6a296eb82cf844ca8539b57c23e6e...
  • zjkC050818
  • zjkC050818
  • 2017-05-06 12:12:26
  • 343

寻找数组中最小的k个数(快排和堆排)

题目描述 输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。 思路1:利用快排的思想,寻找第k个位置上正确的数,k位置...
  • liuguangqiang
  • liuguangqiang
  • 2016-08-15 13:02:57
  • 5552
收藏助手
不良信息举报
您举报文章:【刷题之路】最小的K个数
举报原因:
原因补充:

(最多只允许输入30个字)