（转）基于快速排序的TOPK算法

基于快速排序的TOPK算法

转自：http://blog.csdn.net/fanzitao/article/details/7617223

思想：

类似于快速排序，首先选择一个划分元，如果这个划分元的序号index刚好等于k，那么这个划分元以及左边的数，刚好组成了top-k small data;如果index>k, 那top-k small data在index的左边，那么就继续递归从index-1和数中选取top-k.如果index < k,那么说明还要从index的右边，选取top-(k-index) small data.

代码如下：

public class TopK_Quick {

    public static int Partition(int a[],int low,int high)
    {
        a[0]=a[low];
        int pivokey = a[low];
        while(low<high)
        {
            while(low<high && a[high]>=pivokey) --high;
            a[low] = a[high];
            while(low<high && a[low]<=pivokey) ++low;
            a[high]= a[low];
        }
        a[low]=a[0];
        return low;
    }
    
    public static void display(int a[],int k)
    {
        for(int i=1;i<=k;i++)
        {
            System.out.print(a[i]+" ");
        }
    }
    public static int selectK(int a[],int start,int end,int k)
    {
        int index = 0;
        if(start<end)
        {
            index = Partition(a,start,end);
            if(index == k)//正好找到第k大的数
            {
                index = k;
            }else if(index < k)//还要从index的右边找k-index个数
            {
                index = selectK(a,index+1,end,k-index);
            }else if(index > k)//k个数都在Index的左边
            {
                index = selectK(a,start,index-1,k);
            }
        }
        return index;

    }
    public static void main(String args[])
    {
        int k=0;
        int a[]={0,49,38,29,65,97,76,13,27,49,22,19};
        if(k>0&&k<=a.length-1)
        {
            selectK(a,1,a.length-1,k);
            display(a,k);
        }else{
            System.out.println("Are You Kidding Me?");
        }
        
    }
}

学习之处：

• 二分查找subroutine还可以用于寻找Top K问题。
• 别人代码里面二分查找实现是非递归的！