快速排序,第k小数目(洛谷P1923)
快排代码
#include <iostream>
#include <cstdio>
using namespace std;
int a[100005],n;
void quick_sort(int l,int r)
{
int i=l,j=r,flag = a[l+r>>1];
do{
while(a[i]<flag) i++;
while(a[j]>flag) j--;
if(i<=j){
swap(a[i],a[j]);
i++;j--;
}
}while(i<=j);
if(l<j) quick_sort(l,j);
if(i<r) quick_sort(i,r);
}
int main()
{
cin >> n;
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
}
quick_sort(0,n-1);
for(int i=0;i<n;i++){
cout << a[i] << " ";
}
return 0;
}
第k小数
#include <iostream>
#include <cstdio>
using namespace std;
int a[5000010],n,ans,k;
void findkth(int l,int r)
{
if(l==r){
ans = a[l];
return ;
}
int i=l,j=r,flag = a[l+r>>1];
do{
while(a[i]<flag) i++;
while(a[j]>flag) j--;
if(i<=j){
swap(a[i],a[j]);
i++;j--;
}
}while(i<=j);
if(k<=j) findkth(l,j);
else if(k>=i) findkth(i,r);
else findkth(j+1,i-1);
}
int main()
{
cin >> n >> k;
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
}
findkth(0,n-1);
cout << ans << endl;
return 0;
}
本文来自博客园,作者:TCcjx,转载请注明原文链接:https://www.cnblogs.com/tccjx/articles/16119051.html