package number4;
//在未排序的数组中找到第 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 ≤ 数组的长度。
/**
* 思路:无论数据怎么变换,排序(降序)后第k大的数据都是在第k-1的位置
*
*/
public class Solution5 {
public static void main(String[] args) {
int[] arr= {3,2,3,1,2,4,5,5,6};
int k=4;
System.out.println(findKthLargest(arr,k));
}
public static int findKthLargest(int[] nums, int k) {
return result(nums,0,nums.length-1,k-1);
}
//快速排序算法实现
private static int result(int[] arr, int left, int right,int k) {
if(right==left)//左边等于右边,说明数组只有一个元素,直接返回就好
return arr[left];
int i=left,j=right;
int temp=arr[left];//把左边的数据赋值给临时变量temp
//快速排序的降序排序,大的数据都在左边,小的在右边
while(i<j) {
//找到下标
while(i<j&&arr[j]<temp)
j--;
while(i<j&&arr[i]>=temp)
i++;
//交换数据
if(i<j) {
int t=arr[i];
arr[i]=arr[j];
arr[j]=t;
}
}//循环结束的标志是i==j
arr[left]=arr[i];//这一步很重要,代表着中间的数据是固定住的
arr[i]=temp;
//中间总是已经排好序的数据,现在就来讨论k和左边,右边的关系
//第一种情况,就是k==i的时候,直接返回第k大的元素
if(k==i)
return arr[i];
//第二种情况,k<i,左边就有i个元素
else if(k<i)
return result(arr,left,i-1,k);//这里i-1和i+1的设置,一定程度上减少了数据量
//第三种情况,k>i,
else
return result(arr,i+1,right,k);
}
}
LeetCode_215
最新推荐文章于 2025-05-10 10:39:36 发布