LeetCode

    科技2022-07-17  124

    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); } }
    Processed: 0.015, SQL: 8