快速排序模板题

快速排序

#include<iostream>
using namespace std;
const int N=1e6+10;
int n;
int a[N];

void quicksort(int a[],int l,int r)
{
    if(l>=r) return ;
    
    int x=a[(l+r)>>1],i=l-1,j=r+1;
    while(i<j)
    {
        do i++;while(a[i]<x);
        do j--;while(a[j]>x);
        if(i<j) swap(a[i],a[j]);
    }
    int len=j-r+1;
    if(k)
    quicksort(a,l,j);
    quicksort(a,j+1,r);
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
    }
    quicksort(a,0,n-1);
    
    for(int i=0;i<n;i++) printf("%d ",a[i]);
    return 0;
}

第k个数

#include<bits/stdc++.h>
using namespace std;

const int N=1e5+10;
int n,k;
int a[N];

int quicksort(int l,int r,int k)//对元素下标从l到r,查找第k个元素
{   //快速排序模板
    if(l>=r) return a[r];
    int x=a[(l+r)>>1],i=l-1, j=r+1;//由于比较的时候均是两个指针向前移动的时候再进行比较，所以边界退一个
    while(i<j)
    {
        do i++;while(a[i]<x);
        do j--;while(a[j]>x);
        if(i<j) swap(a[i],a[j]);
    }

    int len = j - l + 1;//元素个数=右边界-左边界+1
    if (k <= len) return quicksort(l, j, k);//不足k个元素时
     return quicksort(j + 1, r, k - len);
}
int main()
{
    scanf("%d %d",&n,&k);
    for(int i=0;i<n;i++)scanf("%d",&a[i]);
    cout<<quicksort(0,n-1,k)<<endl;
    return 0;
}