215. 数组中的第K个最大元素(堆排序忘了,面试翻车。。)

Posted 冲冲冲冲冲冲!!!

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了215. 数组中的第K个最大元素(堆排序忘了,面试翻车。。)相关的知识,希望对你有一定的参考价值。

题目:

思路:使用堆排序选择,时间复杂度可以达到O(nlogk)
(堆排序写法忘了,哭)

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int heapSize = nums.length;
        buildMaxHeap(nums, heapSize);
        for (int i = nums.length - 1; i >= nums.length - k + 1; i--) {
            swap(nums, 0, i);
            heapSize--;
            maxHeapify(nums, 0, heapSize);
        }
        return nums[0];
    }

	/**
	构建堆
	*/
    public void buildMaxHeap(int[] a, int heapSize) {
        for (int i = heapSize / 2; i >= 0; --i) {
            maxHeapify(a, i, heapSize);
        } 
    }
	
	/**
	排序选择
	*/
    public void maxHeapify(int[] a, int i, int heapSize) {
        int l = i * 2 + 1, r = i * 2 + 2, largest = i;
        if (l < heapSize && a[l] > a[largest]) {
            largest = l;
        } 
        if (r < heapSize && a[r] > a[largest]) {
            largest = r;
        }
        if (largest != i) {
            swap(a, i, largest);
            maxHeapify(a, largest, heapSize);
        }
    }

    public void swap(int[] a, int i, int j) {
        int temp = a[i];
        a[i] = a[j];
        a[j] = temp;
    }
}

以上是关于215. 数组中的第K个最大元素(堆排序忘了,面试翻车。。)的主要内容,如果未能解决你的问题,请参考以下文章

[LeetCode]215. 数组中的第K个最大元素(堆)

leetcode215.数组中的第k个最大元素(快排/堆排序 java)***

LeetCode215. 数组中的第K个最大元素

LeetCode.215-数组中的第K个最大元素

215. 数组中的第K个最大元素

leetcode笔记215. 数组中的第K个最大元素