LeetCode——215. 数组中的第K个最大元素
一、题目描述
给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
1.1 示例
示例1
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例2
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
1.2 提示
- 1 <= k <= nums.length <= 104
- -104 <= nums[i] <= 104
二、题目分析
2.1 题目标签:快速排序,分治
2.2 解题思路—分治思想
- 对数组排序后直接返回 nums[nums.length - k] 即可;
- 排序可选择快速排序、归并排序等排序算法;
- 快速排序的具体实现可参考个人博客;
- 时间复杂度:O(nlog2n)。
三、题解——Java
class Solution {
public int findKthLargest(int[] nums, int k) {
quick_sort(nums, 0, nums.length-1);
return nums[nums.length - k];
}
void quick_sort(int[] nums, int l, int r) {
if(l >= r) return;
int i = l - 1, j = r + 1, x = nums[l + (r - l) / 2];
while(i < j) {
do i++; while(nums[i] < x);
do j--; while(nums[j] > x);
if(i < j) {
int t = nums[i];
nums[i] = nums[j];
nums[j] = t;
}
}
quick_sort(nums, l, j);
quick_sort(nums, j+1, r);
}
}