在ACM题中通常会遇到寻找第m小的元素的问题,但是你会发现单纯用排序的方法(sort(),甚至是快排)总是TL超时,下面有一种最快的方法:
需要调用快排中的一次划分:
ACM总结(各类排序的实现)
通过轴值(pivot)的移动,将待排序记录划分成两部分,左侧记录均小于或等于轴值,右侧记录均大于或等于轴值。
调用递归:
1)若pivot==m,则轴值处即为要要找的值;
2)若pivot>m,则说明要寻找的值在pivot的左边(找左边第m小的元素),对数组(1~pivot-1)进行一次划分;
3)若pivot<m,则说明要寻找的值在pivot的右边(找右边第(m-pivot-1)小的元素),对数组(pivot+1~n)进行一次划分;
算法如下:
//寻找第m小的元素
#include<iostream>
#include<algorithm>
#include<vector>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int maxn=200;
int n,m,a[maxn];
int partition(int first,int last){
int i=first,j=last,temp;
while(i<j){
while(i<j&&a[i]<=a[j]){j--;}
if(i<j){
temp=a[i];
a[i]=a[j];
a[j]=temp;
i++;
}
while(i<j&&a[i]<=a[j]){i++;}
if(i<j){
temp=a[i];
a[i]=a[j];
a[j]=temp;
j--;
}}
return i;
}
int find(int first,int last,int m){
int ans=partition(first,last);
if(ans==m){return a[m];}
else if(ans<m){return find(ans+1,last,m);}
else{return find(first,ans-1,m);}
}
int main() {
scanf("%d",&n);
for(int q=1;q<=n;q++){scanf("%d",&a[q]);}
scanf("%d",&m);
printf("%d",find(1,n,m));
return 0;
}