快速排序 第k大数(模板)

快速排序

题目描述
 给定你一个长度为n的整数数列。
 请你使用快速排序对这个数列按照从小到大进行排序。
 并将排好序的数列按顺序输出。
输入格式
 输入共两行，第一行包含整数 n。
 第二行包含 n 个整数（所有整数均在1~10⁹范围内），表示整个数列。
输出格式
 输出共一行，包含 n 个整数，表示排好序的数列。
数据范围
 1≤n≤100000
 1≤n≤100000
输入样例：
 5
 3 1 2 4 5
输出样例：
 1 2 3 4 5

#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
int a[N];

void quick_sort(int l,int r){
    
    if(l >= r) return;
    int i = l - 1, j = r + 1;
    int x = a[l + 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]);
    }
    
    quick_sort(l,j);
    quick_sort(j + 1,r);
}
int main(){
    int n;
    cin >> n;
    for(int i = 0 ; i < n ;++i){
        cin >> a[i];
    }
    quick_sort(0,n - 1);
    for(int i = 0 ; i < n ;++i){
        cout << a[i] << " ";
    }
    return 0;
}

时间复杂度O(nlog_n)，这个模板比较简单，两分钟应该要打完，不过做题也不用他，sort更加好一点，毕竟sort系统做过优化，在数据量小的时候用的插入排序。据说企业面试的时候经常考？

第k个数

题目描述
 给定一个长度为n的整数数列，以及一个整数k，请用快速选择算法求出数列的第k小的数是多少。
输入格式
 第一行包含两个整数 n 和 k。
 第二行包含 n 个整数（所有整数均在1~10⁹范围内），表示整数数列。
输出格式
 输出一个整数，表示数列的第k小数。
数据范围
 1≤n≤100000，1≤k≤n
输入样例：
 5 3
 2 4 1 5 3
输出样例：
 3

参考代码如下，

#include <bits/stdc++.h>
using namespace std;
const int N =100010;
int a[N];

int k_sort(int l,int r,int k){
    if(l >= r) return a[l];
    int x = a[l + r >> 1];
    int 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 s = j - l + 1;
    
    if(k <= s){
        return k_sort(l,j,k);
    }else{
        return k_sort(j + 1, r,k - s);
    }
}
int main(){
    int n,k;
    cin >> n >> k;
    for(int i = 0; i < n ;++i){
        cin >> a[i];
    }
    int ans = k_sort(0, n - 1,k);
    cout << ans << endl;
}

实际上就是单边快排，时间复杂度是O(n)，他很是神奇，不过这个复杂度好像会退化，但是不影响他神奇。