1574.求第k小数
时限:1000ms 内存限制:10000K 总时限:3000ms
描述
求第k小数
输入
先输入一个小于10000的正整数n,再输入n个整数,最后输入一个小于等于n的正整数k,
输出
输出其中第k小的数。
输入样例
5
2 98 34512 8492 1000
2
输出样例
98
解析:这是个线性时间选择题,基本思想是分治和递归。第一次将数组划分成两个数组a[first:mid]和a[mid+1:last],使得前者的每个元素都不大于后者的,然后分别对两者进行相同的操作,划分直到具体的一个数,就使得数组被按从小到大排好了序。基本思想其实跟快速排序没有区别,只是操作略有不同,而且复杂度是O(n),比快排的O(nlogn)更高效。代码如下:
#include <iostream>
using namespace std;
int a[10001];
void swap(int &a,int &b)
{
int temp=a;
a=b;
b=temp;
}
int partition(int a[],int first,int last)
{
int x=a[last];
int middle=first;
int j;
for(j=first;j<last;j++)
{//j从本次传过来的数的第一个到最后一个,如果哪个数小于最后一个数且不是第一个,就把它与第一个数字交换
if(a[j]<x)
{
if(j!=middle)
swap(a[middle],a[j]);
middle++;//交换了之后middle++,结果就成为所有小于最后一个数的所有数都在前面middle个
}
}
swap(a[middle],a[j]);//结束之后,这个区间里从第一个开始前middle个都小于最后一个数,然后交换,最后得到这个区间的序列前middle个是小的,后面的每个数都小于后面的数字
return middle;
}
void select(int a[],int first,int last)
{
if(first<last)//一直分,直到把他们排序完毕
{
int mid=partition(a,first,last);//每次的界线由partitionh函数决定
select(a,first,mid-1);
select(a,mid+1,last);
}
}
int main()
{
int n,k;
cin>>n;
for(int i=0;i<n;i++)
cin>>a[i];
cin>>k;
select(a,0,n-1);
cout<<a[k-1]<<endl;
return 0;
}