基于堆排序的选择方法

最新推荐文章于 2022-09-23 21:25:52 发布

iii_ooo

最新推荐文章于 2022-09-23 21:25:52 发布

阅读量229

点赞数

分类专栏：数据结构与算法文章标签： leetcode 算法排序算法

本文链接：https://blog.csdn.net/iii_ooo/article/details/122582200

版权

数据结构与算法专栏收录该内容

3 篇文章 0 订阅

订阅专栏

文章目录

前言
一、堆排序实现
二、PriorityQueue

前言

求数组中第K个最大元素,源自Letcode

一、堆排序实现

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];

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/kth-largest-element-in-an-array/solution/shu-zu-zhong-de-di-kge-zui-da-yuan-su-by-leetcode-/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

二、PriorityQueue

class Solution{
	public int findKthLargest(int[] nums,int k){
		PriorityQueue<Integer> heap = new PriorityQueue<Integer>((n1,n2)=> n1 - n2);
		
		for(int n:nums){
			heap.add(n);
			if(head.size()>k)
				heap.poll();
		}
		return heap.poll();
	}
}