快速选择算法

简介

在我们求一组元素的第K大值或者前K大值时，可能最先想到的是对元素进行排序，然后选择第K大的或者前K大的值。不过我们只是想取第K大的数，有必须将整组元素进行排序吗？当然不必，这就是我们将要介绍的快速选择算法，其时间复杂度可以达到O(n)。它在一定程度上是基于快速排序的思想。下面我们假设需要寻找一堆元素的中位数。

快速排序

快速排序算基本可以算是比较排序法中效果最优的，它的时间复杂度O(nlogn)。
快速排序步骤：
(1)首先设定一个分界值，通过该分界值将数组分成左右两部分。
(2)将大于或等于分界值的数据集中到数组右边，小于分界值的数据集中到数组的左边。
(3)重复（1）（2）步骤直至数据不能再划分为止。

第一次排序：
以第一个数为分界值，从左往右找一个比分界值大的数，从右往左找一个比分界值小的数，如果比分界值大的在小的左边，则交换两个数位置。再记录下比分界值小的数的个数，最后将分界值放入对应位置，完成第一次排序。接着对分界值左边和右边的数分别进行递归操作直到完成排序。
递归排序：

代码如下：

template<typename T>
static void Qsort(T arr[], int low, int high) {
 //c++;
 if (high <= low) return;
 int i = low;
 int j = high + 1;
 int key = arr[low];
 while (true)
 {
  /*从左向右找比key大的值*/
  while (arr[++i] < key)
  {
   if (i == high) {
    break;
   }
  }
  /*从右向左找比key小的值*/
  while (arr[--j] > key)
  {
   if (j == low) {
    break;
   }
  }
  if (i >= j) break;
  /*交换i,j对应的值*/
  int temp = arr[i];
  arr[i] = arr[j];
  arr[j] = temp;
 }
 /*中枢值与j对应值交换*/
 int temp = arr[low];
 arr[low] = arr[j];
 arr[j] = temp;
 Qsort(arr, low, j - 1);
 Qsort(arr, j + 1, high);
}

选择算法

在前面快速排序中，我们确定了6的位置之后，由于我们需要寻找中位数，完全不需要对6右边的数据进行排序，因此有了选择算法。与快速排序一样，我们将输入数组进行递归划分，与快速排序不同的是，快速排序会递归处理划分的两边，而选择只处理划分一边。
橙色为我们需要确定的中间位置的数（即中位数），黄色为当前作分界值的数，也是一轮比较交换之后能确定位置的数。当我们每次确定位置的数确定的位置在中间位置，那么这个分界值数就是中位数。

代码如下：

template<typename T>
T Select(T arr[], int low, int high, int m) {
 //c++;
 if (high <= low)
 {
  return -1;
 }
 int i = low;
 int j = high + 1;
 int key = arr[low];
 while (true)
 {
  /*从左向右找比key大的值*/
  while (arr[++i] < key)
  {
   if (i == high) {
    break;
   }
  }
  /*从右向左找比key小的值*/
  while (arr[--j] > key)
  {
   if (j == low) {
    break;
   }
  }
  if (i >= j) break;
  /*交换i,j对应的值*/
  int temp = arr[i];
  arr[i] = arr[j];
  arr[j] = temp;
 }
 /*中枢值与j对应值交换*/
 int temp = arr[low];
 arr[low] = arr[j];
 arr[j] = temp;
 if (j > m) {
  mQsort(arr, low, j - 1, m);
 }
 if (j < m) {
  mQsort(arr, j + 1, high, m）);
 }
 if (j = m) {
  return arr[j];
 }
}

在选出第K大的元素之后，我们也自然有了前K大的数，因为在K右边的数都是比K大的，在K左边的数都是比K小的。