力扣(LeetCode) - 215 數(shù)組中的第K個(gè)最大元素

這是一道很經(jīng)典的題目孵延,我在看別人面經(jīng)的時(shí)候不止一次看到了這道題惶凝,因此我們還是很有必要去好好分析一下這道題的。
本題可以用快速排序的partition來解犬钢,或者用堆排序構(gòu)建最小堆來解苍鲜。

一、題目

在未排序的數(shù)組中找到第 k 個(gè)最大的元素玷犹。請注意混滔,你需要找的是數(shù)組排序后的第 k 個(gè)最大的元素,而不是第 k 個(gè)不同的元素歹颓。

示例1:

輸入: [3,2,1,5,6,4] 和 k = 2
輸出: 5

示例2:

輸入: [3,2,3,1,2,4,5,5,6] 和 k = 4
輸出: 4

說明:

你可以假設(shè) k 總是有效的坯屿,且 1 ≤ k ≤ 數(shù)組的長度。

二巍扛、分析

我們可以先對(duì)整體數(shù)據(jù)排序愿伴,然后找出來第k個(gè)數(shù),排序方法我們可以用快速排序电湘、堆排序隔节、歸并排序等鹅经,這樣平均時(shí)間復(fù)雜度是n*log(n)。

2.1 使用最小堆來解

既然讓我們找出第k個(gè)最大的數(shù)怎诫,那么我們是不是可以維護(hù)一個(gè)數(shù)據(jù)結(jié)構(gòu)瘾晃,這個(gè)數(shù)據(jù)結(jié)構(gòu)存儲(chǔ)k個(gè)值,并且是已排序的呢幻妓?我們就想到了堆蹦误。最小堆的父節(jié)點(diǎn)都不大于兩個(gè)子節(jié)點(diǎn)。因此我們可以構(gòu)建一個(gè)最小堆肉津,堆的大小是k强胰;遍歷數(shù)組中的其他元素,如果大于堆頂元素妹沙,則替換偶洋,重新構(gòu)造最小堆;直到所有元素遍歷完距糖,我們得到了這樣一個(gè)最小堆:堆中元素比數(shù)組中其他元素都大玄窝。那么,堆頂元素就是我們要找的第k個(gè)最大的數(shù)字悍引。
時(shí)間復(fù)雜度分析:
最壞情況:數(shù)組已經(jīng)是升序排列了恩脂,時(shí)間復(fù)雜度是n*log(k)
最好情況:數(shù)組已經(jīng)降序排列了,時(shí)間復(fù)雜度是n
平均時(shí)間復(fù)雜度:n

2.2 使用partition

我們回顧一下快速排序的思想趣斤,每次講一個(gè)數(shù)字放到他適合的位置(他小于等于他前面的數(shù)組俩块,并且他大于等于他后面的數(shù)組)。
如果這個(gè)數(shù)字適合的位置恰好是k呢浓领?那么我們不就得到第k大的數(shù)字了嗎玉凯?我們既不需要為他前面的數(shù)字排序,也不需要為他后面的數(shù)字排序镊逝,我們只需要得到第k大的數(shù)字就行壮啊。
時(shí)間復(fù)雜度分析:
最壞情況:數(shù)字是升序排列的,并且我們每次partition選的數(shù)字都是第一個(gè)數(shù)字撑蒜,時(shí)間復(fù)雜度是n*log(n)歹啼。
平均時(shí)間復(fù)雜度:n

三、代碼

java 構(gòu)建最小堆

int[] src = null;
    public int findKthLargest(int[] nums, int k) {
        if (nums == null || nums.length == 0 || k > nums.length)
            return Integer.MAX_VALUE;
        src = nums;
        //構(gòu)造最小堆
        for (int i = (k - 2) / 2; i >= 0; i--) {
            adjustHeap(i, k - 1);
        }
        for (int i = k; i < nums.length; i++) {
            if (nums[i] > nums[0]) {
                swap(i, 0);
                adjustHeap(0, k-1);
            }
        }
        return nums[0];
    }
    
    //構(gòu)建最小堆
    private void adjustHeap(int start, int end) {
        int temp = src[start];
        int j = start * 2 + 1;
        for (j = start * 2 + 1; j <= end; j = j * 2 + 1) {
            if (j < end && src[j] > src[j + 1]) {
                j++;
            }
            if (temp > src[j]) {
                src[start] = src[j];
                start = j;
            } else {
                break;
            }
        }
        src[start] = temp;
    }
    
    private void swap(int i, int j) {
        int temp = src[i];
        src[i] = src[j];
        src[j] = temp;
    }

java partition

    int result = 0;
    int[] src = null;
    int target = 0;
    public int findKthLargest(int[] nums, int k) {
        if (nums == null || nums.length == 0 || k > nums.length)
            return result;
        result = nums[0];
        src = nums;
        target = k-1;
        partition(0,src.length - 1);
        return result;
    }
    
    private void partition (int start, int end) {
        if (start > end)
            return;
        int temp = src[start];
        int i = start;
        int j = end;
        while (i < j) {
            while (i < j && src[j] <= temp) {
                j--;
            }
            if (i < j) {
                src[i] = src[j];
                i++;
            }
            while (i < j && src[i] >= temp) {
                i++;
            }
            if (i < j) {
                src[j] = src[i];
                j--;
            }
        }
        src[i] = temp;
        if (i == target) {
            result = src[i];
            return;
        } else {
            if (i > target) {
                partition(start, i-1);   
            } else {
                partition(i+1, end);                
            }
        }
    }
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末座菠,一起剝皮案震驚了整個(gè)濱河市狸眼,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌浴滴,老刑警劉巖拓萌,帶你破解...
    沈念sama閱讀 218,386評(píng)論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異升略,居然都是意外死亡微王,警方通過查閱死者的電腦和手機(jī)屡限,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,142評(píng)論 3 394
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來炕倘,“玉大人钧大,你說我怎么就攤上這事≌中” “怎么了啊央?”我有些...
    開封第一講書人閱讀 164,704評(píng)論 0 353
  • 文/不壞的土叔 我叫張陵,是天一觀的道長涨醋。 經(jīng)常有香客問我瓜饥,道長,這世上最難降的妖魔是什么浴骂? 我笑而不...
    開封第一講書人閱讀 58,702評(píng)論 1 294
  • 正文 為了忘掉前任乓土,我火速辦了婚禮,結(jié)果婚禮上靠闭,老公的妹妹穿的比我還像新娘帐我。我一直安慰自己坎炼,他們只是感情好愧膀,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,716評(píng)論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著谣光,像睡著了一般檩淋。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上萄金,一...
    開封第一講書人閱讀 51,573評(píng)論 1 305
  • 那天蟀悦,我揣著相機(jī)與錄音,去河邊找鬼氧敢。 笑死日戈,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的孙乖。 我是一名探鬼主播浙炼,決...
    沈念sama閱讀 40,314評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼唯袄!你這毒婦竟也來了弯屈?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,230評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤恋拷,失蹤者是張志新(化名)和其女友劉穎资厉,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體蔬顾,經(jīng)...
    沈念sama閱讀 45,680評(píng)論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡宴偿,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,873評(píng)論 3 336
  • 正文 我和宋清朗相戀三年湘捎,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片窄刘。...
    茶點(diǎn)故事閱讀 39,991評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡消痛,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出都哭,到底是詐尸還是另有隱情秩伞,我是刑警寧澤,帶...
    沈念sama閱讀 35,706評(píng)論 5 346
  • 正文 年R本政府宣布欺矫,位于F島的核電站纱新,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏穆趴。R本人自食惡果不足惜脸爱,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,329評(píng)論 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望未妹。 院中可真熱鬧簿废,春花似錦、人聲如沸络它。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,910評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽化戳。三九已至单料,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間点楼,已是汗流浹背扫尖。 一陣腳步聲響...
    開封第一講書人閱讀 33,038評(píng)論 1 270
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留掠廓,地道東北人换怖。 一個(gè)月前我還...
    沈念sama閱讀 48,158評(píng)論 3 370
  • 正文 我出身青樓,卻偏偏與公主長得像蟀瞧,于是被迫代替她去往敵國和親沉颂。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,941評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容

  • 概述 排序有內(nèi)部排序和外部排序黄橘,內(nèi)部排序是數(shù)據(jù)記錄在內(nèi)存中進(jìn)行排序兆览,而外部排序是因排序的數(shù)據(jù)很大,一次不能容納全部...
    蟻前閱讀 5,184評(píng)論 0 52
  • 1.插入排序—直接插入排序(Straight Insertion Sort) 基本思想: 將一個(gè)記錄插入到已排序好...
    依依玖玥閱讀 1,254評(píng)論 0 2
  • 1 初級(jí)排序算法 排序算法關(guān)注的主要是重新排列數(shù)組元素塞关,其中每個(gè)元素都有一個(gè)主鍵抬探。排序算法是將所有元素主鍵按某種方...
    深度沉迷學(xué)習(xí)閱讀 1,408評(píng)論 0 1
  • 排序(下):如何用開排思想在O(n)內(nèi)查找第K大元素 上一節(jié)我講了冒泡排序、插入排序、選擇排序這三種排序算法小压,它們...
    GhostintheCode閱讀 822評(píng)論 0 0
  • 肖央一醒來,就發(fā)現(xiàn)自己蹲在50多米高的酒店房間窗外蜻牢,旁邊白色的空調(diào)外機(jī)隆隆作響烤咧,吹著自己白色的浴袍不停的扇動(dòng),他只...
    曉二姑娘閱讀 726評(píng)論 5 9