TopK问题

TopK问题

今天看完了排序，想到了之前写过的一道题，输出前K大元素。
有两种方法，这里只要介绍一种基于快排的方法。

快排

由于我们只需要取前K大，所以不需要多余的元素，我们容易利用快排的性质，选取 主元$pivot$之后，如果

1）比$pivot$大的元素数目之和等于$k$，那么停止算法，位于$pivot$右边的元素就是我们所要的。
2）比$pivot$大的元素数目之和大于$k$，那么继续递归去找$pivot$左边的前$k-m$个元素, $m$是包括$pivot$自身和其右边所有元素的数目。
3）比$pivot$大的元素数目之和小于$k$，那么继续递归去找$pivot$右边元素的前$K$个元素。

原问题形式为: $arrangeRight(left,right,k)$
下面对应三个子问题：($i$是$pivot$的位置)
1) 结束算法。
2) $arrangeRight(left,i,k-m)$
3) $arrangeRight(i,right,k)$

这里借用一下郭炜老师的PPT哈哈（我爱郭老师）

平均复杂度分析
$m$是常数
$T(N)=T(N/2)+m\ast N$
$=T(N/4)+m\ast N+m\ast N/2$
$=T(N/8)+m\ast N+m\ast N/2+m\ast N/4$
$=......$
$=T(1)+m\ast N+m\ast N/2+m\ast N/4+...$
$<2\ast m\ast N$

所以算法的平均时间复杂度是$O(N)$

代码

题目链接（openjudge）
之前写的代码了，感觉有些地方写的不太好。

#include<iostream>
#include<algorithm>
#define maxsize 100010
using namespace std;
int a[maxsize], n, k;

void arrangeRight(int a[],int left,int right,int k)
{
    if(left>=right)	return;
    int i=left,j=right,temp=a[i];
    while(i!=j)
    {
        while(i<j&&a[j]>=temp) j--;
        swap(a[i],a[j]);
        while(i<j&&a[i]<temp)  i++;
        swap(a[i],a[j]);
    }
    if (right-i+1==k)  return;
    else if (right-i+1>k)
        arrangeRight(a,i+1,right,k);
    else if (right-i+1<k)
        arrangeRight(a,left,i-1,k-(right-i+1));
}

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

堆排序

思想很简单，就是维护一个只有$k$个元素的最小堆，扫描一遍所有元素。每次遇到一个元素如果比堆顶元素大用这个元素替换掉堆顶元素。
最后留在堆里的就是答案。

复杂度分析
遍历一遍所有元素，复杂度$O(N)$。
那么在遍历的时候每次对堆的操作的复杂度呢？首先我们可以确定，对堆的操作复杂度是$log(K)$，$K$是堆中元素的数量，那么这就是一个平均的操作复杂度。在$K$相对于$N$很小的情况下，这个就可以忽略不计，也就是说复杂度还是$O(N)$，不过堆的话常数可能会大一点。