Acwing 基础课—基础算法—快速排序

快速排序模板

参考链接

https://www.acwing.com/solution/content/2096/
核心思想就是快排的每一趟，数轴的左边都会是 <= x 的， 右边都是 >= x 的。

#include<bits/stdc++.h>
using namespace std;
int a[100005];

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

2 第k小的数

题目链接

https://www.acwing.com/problem/content/788/

题目大意

给定一个长度为n的整数数列，以及一个整数k，请用快速选择算法求出数列从小到大排序后的第k个数。

解题思路

快排的每一趟，数轴的左边都会是 <= x 的， 右边都是 >= x 的。
左边元素的个数是 s1 = j - l + 1, 如果k <= s1 的话，那么下次递归的区间就是左边，否则右边。
直到 l == r 时返回a[l]。

#include<bits/stdc++.h>
using namespace std;
int a[100005];

//第k小的数
int qs(int l,int r,int k){
    if(l == r) return a[l];
    int i = l - 1,j = r + 1;
    int x = a[l];
    while(i < j){
        while(a[--j] > x);
        while(a[++i] < x);
        if(i < j)
            swap(a[i],a[j]);
    }
    int s1 = j - l + 1;
    if(k <= s1)
        qs(l,j,k);
    else
        qs(j+1,r,k-s1);
}

int main(){
    int n,k;
    cin >> n >> k;
    for(int i = 0;i < n;i++) cin >> a[i];
    cout << qs(0,n - 1,k) << endl;
    return 0;
}