无序数组求第K大/第K小的数

方法一：quicksort

根据快排思想，从后往前找比基准数小的，交换位置。

从前往后找比基准数大的，交换位置。

最后安放基准数。

保证 l到p 是大数，若 p-l+1==k 那么p就是第K大

若 p-l+1<k 那么从 p+1 到 r 中 找 k-（p-l+1）大的数

若 p-l+1>k 那么从 l 到 p-1中 找第k大的数。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n, k;
int a[100010];
int par(int l, int r)
{
    int t = a[l];
    while (l < r)
    {
        while (l < r &&a[r] <= t)    r--;
        a[l] = a[r];
        while (l < r &&a[l] >= t)    l++;
        a[r] = a[l];
    }
    a[l] = t;
    return l;
}
int search(int l, int r,int k)
{
    if (l <= r)
    {
        int p = par(l, r);
        if (p - l + 1 == k)    return p;
        else if (p - l + 1 < k)    return search(p+1,r,k-(p-l+1));
        else return search(l,p-1,k);
    }
}
int main()
{
    cin >> n >> k;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    cout << a[search(1, n, k)] << endl;

    return 0;
}

 

方法二：最小堆

原本用swap函数的，后来好像swap并不能使两个数交换，以后我一定自己写。（鞠躬表示歉意）

搞了半天搞错了，还是挖坑吧。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<map>
#include<algorithm>
#include<cmath>
using namespace std; 
int n, a[11000],b,k;
void adjust(int i,int n)
{
    int child = 2 * i;
    while (child <= n)
    {
        if (child + 1 <= n && a[child] < a[child + 1])
            child++;
        if (a[child] < a[i])
            break;
        int t = a[child];
        a[child] = a[i];
        a[i] = t;
        i = child;
        child = 2 * i;
    }
}
void dui(int n)
{
    for (int i = n / 2; i >= 1; i--)
        adjust(i, n);
}
int main()
{
    cin >> n >> k;
    for (int i = 1; i <= k; i++)
        cin >> a[i];
    dui(k);
    for (int i = 1; i <= n - k; i++)
    {
        cin >> b;
        if (b < a[1])
        {
            a[1] = b;
            adjust(1,k);
        }
    }
    cout << a[1] << endl;
    return 0;
}