基于快排的partition实现TOPK问题

 

类似于快速排序，首先选择一个划分元，如果这个划分元的序号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.

1. public class TopK_Quick {
3. public static int Partition(int a[],int low,int high)
4. {
5. a[0]=a[low];
6. int pivokey = a[low];
7. while(low<high)
8. {
9. while(low<high && a[high]>=pivokey) --high;
10. a[low] = a[high];
11. while(low<high && a[low]<=pivokey) ++low;
12. a[high]= a[low];
13. }
14. a[low]=a[0];
15. return low;
16. }
18. public static void display(int a[],int k)
19. {
20. for(int i=1;i<=k;i++)
21. {
22. System.out.print(a[i]+" ");
23. }
24. }
25. public static int selectK(int a[],int start,int end,int k)
26. {
27. int index = 0;
28. if(start<end)
29. {
30. index = Partition(a,start,end);
31. if(index == k)//正好找到第k大的数
32. {
33. index = k;
34. }else if(index < k)//还要从index的右边找k-index个数
35. {
36. index = selectK(a,index+1,end,k-index);
37. }else if(index > k)//k个数都在Index的左边
38. {
39. index = selectK(a,start,index-1,k);
40. }
41. }
42. return index;
44. }
45. public static void main(String args[])
46. {
47. int k=0;
48. int a[]={0,49,38,29,65,97,76,13,27,49,22,19};
49. if(k>0&&k<=a.length-1)
50. {
51. selectK(a,1,a.length-1,k);
52. display(a,k);
53. }else{
54. System.out.println("Are You Kidding Me?");
55. }
57. }
58. }