TopK (MinK) 实现

概述：基于快排原理找到最小的K个元素，属于Top K问题。注意，使用快排原理找前K小问题不需要对整个数组进行O(nlogn)的排序。我们只要找K所在的区间进行递归调用，即每次只要对数据的一半进行递归调用，而不用对全部的数据进行递归调用，从而降低了时间复杂度。平均情况下，时间复杂度计算：n + n/2 +  ... + n/n = 2n, 因此时间复杂度是O(n)。

 import java.util.Arrays;
 public class MinK {
     public static void main(String[] args) {
         int[] arr = {3,45,78,64,52,64,64,55,55,11,2};
         System.out.println(arrayToString(arr,"Array:"));
         int k = 3;
         int[] res = quickSearch(arr, 0, arr.length-1, k);
         System.out.println(arrayToString(res,"MinK :"));
     }
     /**
      * 基于快排找到最小的K个元素
      * nums   数组
      * left  数组左下标
      * right 数组右下标
      * k     k值
      */

     private static int[] quickSearch(int[] nums, int left, int right, int k) {
         // 每快排切分1次，找到排序后下标为j的元素，如果j恰好等于k就返回j以及j左边所有的数；
         int j = partition(nums, left, right);
         if (j == k) {
             return Arrays.copyOf(nums, j);
         }
         else if (j < k) {
             return quickSearch(nums,j+1, right, k);
         }
         else{
             return quickSearch(nums, left, j-1, k);
         }
     }

     // 快排切分，返回下标j，使得比numss[j]小的数都在j的左边，比numss[j]大的数都在j的右边。
     private static int partition(int[] nums, int left, int right) {
         int key = nums[left];
         int i = left;
         int j = right;
         while(i<j){   //当i=j时退出总循环
             while(nums[j] >= key && i < j) j--;  //从右往左，找到比key小的值方可退出循环
             while(nums[i] <= key && i < j) i++;  //从左往右，找到比key大的值方可退出循环
             if(i<j){
                 int temp = nums[i];  // 把左边大的值与右边小的值进行交换
                 nums[i] = nums[j];
                 nums[j] = temp;
             }
         }
         nums[left] = nums[i];
         nums[i] = key;   //把key交换到中间
         return i;
     }
     private static String arrayToString(int[] arr,String flag) {
         String str = flag+'\t';
         for(int a : arr) {
             str += a + "\t";
         }
         return str;
     }

 }

结果：