#include<iostream>
#include<cmath>
using namespace std;
#define NUM 1001
int a[NUM];
int select(int left, int right, int k)//left:数组起始下标(本题从0开始);right:终点下标;k:数组中第k小的数。
{
int i = left;//i相当于指针(从左向右)
int j = right;//j相当于指针(从右向左)
int pivot = a[left];//选第一个数作为分界数据,将比它小的数放在它左边,大的放在右边。
while (true)
{
while (a[i] < pivot)//如果a[i] < pivot,i指针就不停的向右移动
{
i++;
}while (a[j] > pivot)//如果a[j] > pivot, j指针就不停的向左移动
{
j--;
}
if (i == j) break;//如果i==j说明将pivot放入了最中间的位置。
else
{
swap(a[i], a[j]);//走不动了,交换一下a[i]和a[j]。
}
if (a[i] == a[j])j--;//如果遇到两数相等,默认j所指的大一点,j--。如(2 1 2 5 3) 如果没这一步会死循环。
}
//nleft表示pivot左边的元素个数。nleft=j-left。
if (j - left == k - 1)//pivot的位置等于左边的元素个数+1.这时正好直接返回pivot。
{
return pivot;
}
if (j - left < k - 1)//右边多了,从j+1开始向右寻找第k-(nleft+1)小的数。(因为已经找到了前nleft+1小的数了。)
{
return select(j + 1, right, k - (j - left + 1));
}
else//说明第k小的数在左边集合中,从第一个元素开始找,找到第k小的元素就好。
{
return select(left, j - 1, k);
}
}
int main()
{
int n,x;
cout << "请输入有几个数" << endl;
cin >> n;
cout << "请输入要寻找第几小的数" << endl;
int k;
cin >> k;
for (int i = 0; i < n; i++)
{
cout << "请输入第" << i+1 << "个数的值是多少" << endl;
cin >> x;
a[i] = x;
}
cout<<"第"<<k<<"小的数是" << select(0, n - 1, k);
}