当前位置:   article > 正文

解决TOP-K问题:快速选择排序算法_cuda 求第k大的值

cuda 求第k大的值

1. 思想
   Quick select算法通常用来在未排序的数组中寻找第k小/第k大的元素。
   Quick select和Quick sort类似,核心是partition。 

   1. 什么是partition?

   从数组中选一个数据作为pivot,根据每个数组的元素与该pivot的大小将整个数组分为两部分:
   左半部分,都比pivot大,右半部分,都比pivot小 。

  2. 用分治思路实现排序

   pivotIndex 是pivot在数组的下标

   pivotIndex大于k,说明array[pivotIndex]左边的元素都大于k,只递归array[0, pivotIndex-1]第k大的元素即可;

   pivotIndex小于k,说明第k大的元素在array[pivotIndex]的右边,只递归array[pivotIndex +1, n]第k-pivotIndex大的元素即可;

代码:

  1. @Test
  2. public void main() {
  3. int[] arr = {3,6,2,8,0,7,10};
  4. int k = 5;
  5. quickSelectSort(arr, 0, arr.length-1, k);
  6. for (int i=0; i<k; i++) {
  7. System.out.println(arr[i]);
  8. }
  9. }
  10. public void quickSelectSort(int[] arr, int low, int high, int k) {
  11. int index = quickSort(arr, low, high);
  12. if (k == index-low+1) {
  13. } else if (k < index-low+1) {
  14. quickSelectSort(arr, low, index-1, k);
  15. } else if (k > index-low+1) {
  16. quickSelectSort(arr, index+1, high, k-(index-low+1));
  17. }
  18. }
  19. public int quickSort(int[] arr, int i, int j) {
  20. int temp = arr[i];
  21. while (i < j) {
  22. while (temp <= arr[j] && i<j) {
  23. j--;
  24. }
  25. arr[i] = arr[j];
  26. while (temp >= arr[i] && i<j) {
  27. i++;
  28. }
  29. arr[j] = arr[i];
  30. }
  31. arr[i] = temp;
  32. return i;
  33. }

与Quick sort不同的是,Quick select只考虑所寻找的目标所在的那一部分子数组,而非像Quick sort一样分别再对两边进行分  割。正是因为如此,Quick select将平均时间复杂度从O(nlogn)降到了O(n)。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/716762
推荐阅读
相关标签
  

闽ICP备14008679号