使用快排解決top k的問題 解法是最優的
時間復雜度 O(N)，空間復雜度 O(1)

public class Qs {public static void main(String[] args) {// TODO Auto-generated method stubSystem.out.println("dss");int array[]= {3,2,1,2,5};System.out.println(findKlargest(array,4));}public static int findKlargest(int array[],int k) {k=array.length-k;int l=0;int r=array.length-1;while(l<r) {int j=parition(array,l,r);if(j==k) {break;}else if(j>k) {r=j-1;}else if (j<k) {l=j+1;}}return array[k];}public static int parition(int array[],int l,int r) {int i=l;int j=r+1;while(true) {while(array[++i]<array[l]&&i<r);while (array[--j]>array[l]&&j>l);if(i>=j) {break;}swap(array, i, j);}swap(array, l, j);return j;}private static void swap(int[] a, int i, int j) {int t = a[i];a[i] = a[j];a[j] = t;} }

解析
主要包含三個函數，主函數找到最大k， 循環l<r 然后判斷partition 返回來的是不是k，進行判斷
快排函數選取第一個作為pivot，進行對比 交換 返回該游標 的位置

快排

public class QuickSort<T extends Comparable<T>> extends Sort<T> {@Overridepublic void sort(T[] nums) {shuffle(nums);//先洗牌sort(nums, 0, nums.length - 1);}private void sort(T[] nums, int l, int h) {if (h <= l)return;int j = partition(nums, l, h);sort(nums, l, j - 1);sort(nums, j + 1, h);}private void shuffle(T[] nums) {List<Comparable> list = Arrays.asList(nums);Collections.shuffle(list);list.toArray(nums);} } private int partition(T[] nums, int l, int h) {int i = l, j = h + 1;T v = nums[l];while (true) {while (less(nums[++i], v) && i != h) ;while (less(v, nums[--j]) && j != l) ;if (i >= j)break;swap(nums, i, j);}swap(nums, l, j);return j; }

總結

以上是生活随笔為你收集整理的快速排序之topk的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。