欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【SSL_1421】求第K小数

程序员文章站 2022-05-14 20:19:57
...

求第K小数

Description

于给定n个元素的数级a[1…n],要从中找出第k小的元素。

Input

第一行是总数n和k,第二行是n个待比较的数。

Output

第K小数的位置。

Sample Input

5 3
23 8 91 56 4

Sample Output

1


题解

用快排,在快排的过程中求第k小。
因为快排每次都会把比key小的的数放在左边,把比key大的数放在右边
所以每次排完后,若key在第 i i i个位置,则key是第 i i i小的数。
若: i = k i=k i=k则找到第 k k k小的数,输出; i < k i<k i<k则只用往右递归; i > k i>k i>k则只用往左递归

代码

#include<queue>
#include<iostream>
#define ll unsigned int //unsigned 是为了抢最优解
#define M 1000001
using namespace std;
struct node
{
	ll num,id; //num是数,id是位置
} a[M];
ll n;
void qsort(ll k,ll l,ll r)
{
	ll i=l,j=r,key=a[((l+r)>>1)].num;
	if(i>j) return;
	while(i<j)
	{
		while(a[i].num<key&&i<j) i++; //从左往右找比key大的数
		while(a[j].num>key&&j>i) j--; //从右往左找比key小的数
		swap(a[i],a[j]);
	}
	if(k==i) cout<<a[i].id; //找到第k小的数
	if(k<i) qsort(k,l,i-1); //往左递归
	if(k>i) qsort(k,i+1,r); //往右递归
}
int main()
{
	ll k;
	scanf("%d%d",&n,&k);
	for(ll i=1; i<=n; a[i].id=i,i++) scanf("%d",&a[i].num);
	qsort(k,1,n);
}