LeetCode – Kth Largest Element in an Array (Java)

最新推荐文章于 2024-11-19 02:25:22 发布

翻译最新推荐文章于 2024-11-19 02:25:22 发布 · 290 阅读

leetcode刷题同时被 2 个专栏收录

72 篇文章

订阅专栏

java

8 篇文章

订阅专栏

本文介绍了三种不同的方法来寻找数组中的第K大元素：直接排序、快速选择和最小堆。详细展示了每种方法的实现过程及代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

通过排序

public int findKthLargest(int[] nums, int k) {
    Arrays.sort(nums);
    return nums[nums.length-k];
}

同快速排序方法差不多

public int findKthLargest(int[] nums, int k) {
    if (k < 1 || nums == null) {
        return 0;
    }

    return getKth(nums.length - k +1, nums, 0, nums.length - 1);
}

public int getKth(int k, int[] nums, int start, int end) {

    int pivot = nums[end];

    int left = start;
    int right = end;

    while (true) {

        while (nums[left] < pivot && left < right) {
            left++;
        }

        while (nums[right] >= pivot && right > left) {
            right--;
        }

        if (left == right) {
            break;
        }

        swap(nums, left, right);
    }

    swap(nums, left, end);

    if (k == left + 1) {
        return pivot;
    } else if (k < left + 1) {
        return getKth(k, nums, start, left - 1);
    } else {
        return getKth(k, nums, left + 1, end);
    }
}

public void swap(int[] nums, int n1, int n2) {
    int tmp = nums[n1];
    nums[n1] = nums[n2];
    nums[n2] = tmp;
}

通过堆

public int findKthLargest(int[] nums, int k) {
    PriorityQueue<Integer> q = new PriorityQueue<Integer>(k);
    for(int i: nums){
        q.offer(i);

        if(q.size()>k){
            q.poll();
        }
    }

    return q.peek();
}