设计一个平均时间为O(n)的算法,在n(1<=n<=1000)个无序的整数中找出第k小的数。
提示:函数int partition(int a[],int left,int right)的功能是根据a[left]a[right]中的某个元素x(如a[left])对a[left]a[right]进行划分,划分后的x所在位置的左段全小于等于x,右段全大于等于x,同时利用x所在的位置还可以计算出x是这批数据按升非降序排列的第几个数。因此可以编制int find(int a[],int left,int right,int k)函数,通过调用partition函数获得划分点,判断划分点是否第k小,若不是,递归调用find函数继续在左段或右段查找。
输入格式:
输入有两行:
第一行是n和k,0<k<=n<=10000
第二行是n个整数
输出格式:
输出第k小的数
输入样例:
在这里给出一组输入。例如:
10 4
2 8 9 0 1 3 6 7 8 2
结尾无空行
输出样例:
在这里给出相应的输出。例如:
2
结尾无空行
知识点:
- 快排,剪枝
思路:
- 对序列进行快排,
- s=j-left+1,即排列左边元素个数,当s>=k即对序列左边进行排序即可,
- 当s<k即要找的数在序列的右边即对序列的右边进行排序即可当时要找的数k=k-s(即要找的数要减去序列左边的个数,即在右边第k-s个)
- 当
if(a[i]>=a[j]) return a[i]
即找到第k个,结束寻找返回a[i]。
注意:
- 当
if(a[i]>=a[j]) return a[i]
即找到第k个,结束寻找返回a[i]。
源码:
#include<bits/stdc++.h>
using namespace std;
int n,k;
int a[10005];
int fun(int left,int right,int k){
int s;
int i=left;
int j=right;
int step=a[i];
if(i>=j)
return a[i];
while(i<j){
while(j>i&&a[j]>=step)
j--;
while(i<j&&a[i]<=step)
i++;
if(i<j){
int temp;
temp=a[i];
a[i]=a[j];
a[j]=temp;
}
}
a[left]=a[i];
a[i]=step;
s=j-left+1;
if(s>=k){
fun(left,i,k);
}
else{
fun(i+1,right,k-s);
}
}
int main(){
cin>>n>>k;
for(int i=1;i<=n;i++){
cin>>a[i];
}
cout<<fun(1,n,k);
}