线性选择也就是类似于快排得方式,每次确定一个位置,左边的数都比选择这个小(大)右边都比它大(小)。当选择第k小的数字,当在右边部分,就去递归右边部分舍去左边,(在左边舍去右边)
里面的sort是快速排序
select是线性选择
#include<iostream>
using namespace std;
#include<time.h>
#include<stdlib.h>
//引用传值
void swap(int &a,int &b){
int temp = a;
a=b;
b = temp;
}
int Partition(int a[], int p, int r) {
int i = p, j = r;
int x = a[p];
while (i<j)
{
while (a[j]>=x && i<j)
j--;
while (a[i]<=x&&i<j)
i++;
if(i<j)
swap(a[i], a[j]);
}
swap(a[p],a[j]);
return j;
}
int RP(int a[], int p, int r) {
srand(time(NULL));
int i = rand() % (r - p +1) + p;
swap(a[i], a[p]);
return Partition(a, p, r);
}
int select(int a[], int p, int r, int k) {
if (p == r) return a[p];
int i = RP(a, p, r);
int j = i - p + 1;
if (k <= j) return select(a, p, i, k);
else return select(a, i + 1, r, k - j);
}
void sort(int a[],int p,int r){
int i = RP(a, p, r);
if(i>p)
sort(a,p,i-1);
if(i<r)
sort(a,i+1,r);
}
int main()
{
int n;
cin >> n;
int m;
cin >> m;
int *arr = new int[n];
cin.get();
for (int k = 0; k < n; k++)
{
cin >> arr[k];
}
int ans = select(arr,0,n-1,m);
cout<<ans;
return 0;
}