数组中最小的k个数字【排序,最大堆,Quick Select算法】

题目:

输入n个数,找出其中最小的k个数(k<=n)。

思路:

思路一:

将n个数排序,输出前k个。时间复杂度为O(nlogn)。当k比较大的时候(比如k>n/2),这种方法是一种不错的方法,但是在k比较小的时候,我们并不关心k位之后的数字的顺序,我们甚至也不关心前k个数字的顺序,这种方法带来了额外的操作,所以很显然时间上还有额外的优化空间。

思路二:

维护一个容量为k最大堆,将n个数入堆,最后堆中剩的k个数字就是最小的k个数。维护堆的时间复杂度为logk,需要将n个数字依次入堆,所以这种方法的时间复杂度为O(nlogk)。

思路三:

在方法二中,维护堆的过程依然可以得到前k个数字的顺序,而我们是不关心前k个数字的顺序的。用什么办法可以把数组分成两部分,而不关心每一部分的顺序呢?

当我们在进行快排的时候,我们每次会选择一个“哨兵”,小于哨兵的放在左边,大于哨兵的放在右边,那么如果哨兵的位置恰好为k,那不就刚好分好了吗?如果哨兵的位置不那么完美怎么办呢?递归进行下一次划分就好了!如果哨兵的位置>k,那么就递归处理左边的部分,否则处理右边的部分。

这种方法被称为“Quick Select”,与快排相比,快排需要每次递归处理两边的区间,而这种方法只处理一边,所以时间复杂度低于快排。利用主定理可以求出时间复杂度为O(n),在不关心前k个数字顺序时是最高效的算法之一。

代码:

import java.util.*;

public class Solution {

    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {

        ArrayList<Integer> ans = new ArrayList<>();

        if (k>input.length) {

        quickSelect(input,0,input.length-1,k-1);

        for (int i = 0;i<k;i++){

            ans.add(input[i]);

        }

        return ans;

    }

    public void quickSelect(int[] a,int l,int r,int k){

        if (l>=r) return;

        if (r == l+1){

            if (a[l]>a[r]){

                swap(a,l,r);

            }

            return;

        }

        int i = l;

        int j = r+1;

        int tmp = 0;

        while (i<j){

            i++;

            while (i<=r&&a[i]<a[l]) i++;

            j--;

            while (j>l&&a[j]>a[l]) j--;

            if (i<j) swap(a,i,j);

        }

        swap(a,l,j);

        if (j == k){

            return;

        }else if (j>k){

            quickSelect(a,l,j-1,k);

        }else{

            quickSelect(a,j+1,r,k);

        }

    }

    public void swap(int[] a,int x,int y){

        int tmp = a[x];a[x] = a[y];a[y] = tmp;

    }

}

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,634评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,951评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,427评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,770评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,835评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,799评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,768评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,544评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,979评论 1 308
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,271评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,427评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,121评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,756评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,375评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,579评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,410评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,315评论 2 352

推荐阅读更多精彩内容

  • 在C语言中,五种基本数据类型存储空间长度的排列顺序是: A)char B)char=int<=float C)ch...
    夏天再来阅读 3,340评论 0 2
  • "use strict";function _classCallCheck(e,t){if(!(e instanc...
    久些阅读 2,028评论 0 2
  • 总结一下常见的排序算法。 排序分内排序和外排序。内排序:指在排序期间数据对象全部存放在内存的排序。外排序:指在排序...
    jiangliang阅读 1,338评论 0 1
  • 文/砚蓝 走在大街上,商铺小摊儿音响海报,到处都在向行人抛着媚眼,让人目不暇接,陷入选择性困难的迷乱。而我更爱看美...
    帮得上阅读 408评论 0 0
  • 重庆市聋协组织重阳节活动圆满结束了!我认识几个老聋人,和他们聊聊开心。见到了从前的我同事,和她唠家常。活动的开始时...
    WR月光女神阅读 156评论 0 1