215. 数组中的第K个最大元素
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:
你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。
class Solution {
public int findKthLargest(int[] nums, int k) {
if(nums.length < 2){
return nums.length == 1 ? nums[0] : -1;
}
return countingSort(nums, k);
}
private int countingSort(int[] nums, int k){
int max = 0;
int min = 0;
for(int i = 0; i < nums.length; i++){
if(nums[i] > max){
max = nums[i];
}
if(nums[i] < min){
min = nums[i];
}
}
int length = (max - min) + 1;
int[] newArray = new int[length];
//这个数组记录的是与最小的值相差的某位的数量
for(int i = 0; i < nums.length; i++){
newArray[nums[i] - min]++;
}
int j = 0;
for(int i = newArray.length - 1; i >= 0; i--){
//这样最大值就出来了,先从最大值开始
if(newArray[i] > 0){
j = newArray[i] + j;
if(j >= k){
return i + min;
}
}
}
return -1;
}
}
215. 数组中的第K个最大元素
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:
你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。
class Solution {
public int findKthLargest(int[] nums, int k) {
if(nums.length < 2){
return nums.length == 1 ? nums[0] : -1;
}
return countingSort(nums, k);
}
private int countingSort(int[] nums, int k){
int max = 0;
int min = 0;
for(int i = 0; i < nums.length; i++){
if(nums[i] > max){
max = nums[i];
}
if(nums[i] < min){
min = nums[i];
}
}
int length = (max - min) + 1;
int[] newArray = new int[length];
//这个数组记录的是与最小的值相差的某位的数量
for(int i = 0; i < nums.length; i++){
newArray[nums[i] - min]++;
}
int j = 0;
for(int i = newArray.length - 1; i >= 0; i--){
//这样最大值就出来了,先从最大值开始
if(newArray[i] > 0){
j = newArray[i] + j;
if(j >= k){
return i + min;
}
}
}
return -1;
}
}