设计一个平均时间为 O ( n ) O(n) O(n)的算法,在 n ( 1 < = n < = 1000 ) n(1<=n<=1000) n(1<=n<=1000)个无序的整数中找出第k小的数。
因为本题要求使用
O
(
n
)
O(n)
O(n)的时间,所以不能直接采用排序然后输出的方法来解题。
因此采用分治方法,先任意找数组中的一个元素
a
a
a(代码中的
a
a
a 为数组的第一个元素,亦可才用随机数选取数组中的任一个元素),采用快速排序将数组进行一次划分,即将小于a的元素放在其左侧,大于
a
a
a 的元素放在其右侧。
然后判断元素
a
a
a 是否满足题目为第k小的数,满足则直接输出,否则判断下一次在哪一区间进行划分。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int patition(int a[], int left, int right){ //对 left ~ right 进行划分
int x = a[left];
while(left < right){
while(left<right && a[right]>=x) right--;
a[left] = a[right];
while(left<right && a[left]<=x) left++;
a[right] = a[left];
}
a[left] = x;
return left;
}
void find(int a[], int left, int right, int k){ //在 left ~ right 寻找
int pos = patition(a, left, right);
if(pos == k) printf("%d\n", a[pos]);
else if(pos > k) find(a, left, pos-1, k);
else find(a, pos+1, right, k);
}
int main(){
int n, k, a[10005];
scanf("%d%d", &n, &k);
for(int i=1; i<=n; i++) scanf("%d", a+i);
find(a, 1, n, k);
}