Quick-Selection问题描述:
Goal: 给了一个有N个元素的数组,找出最大的第K个元素。(或返回最大or最小的K个)
Ex: Min(k = 0), Max(k = N-1), Medium(k = N/2)
思路:
给数组Partition(如果是返回最大or最小的K个元素不一定要排序,因为没要求返回排好序的K个数,但Partition用的是Quick Sort的),找到第k个最大的数。
1. Partition数组使得:
- a[j]
的位置已归位
- 位置j
左边的数都比它小
- 位置j
右边的数都比它大
2. 判断当前归位的j
是否等于k,若K<j
,则partitionj
左边的部分,若k>j
,则partitionj
右边的部分,由此递归,直到j
等于k,算法结束。(如果在第一轮partition结束后j就刚好等于k则算法结束。)
Language: Java
public static Comparable select(Comparable[] a, int k)
{
StdRandom.shuffle(a);
while(hi > lo)//当没有return且hi>lo时继续执行while,继续partition
{
int j = partition(a, lo, hi);
if(j < k)
lo = j+1;//下一次partition j 右边的subarray
else if(j > k)
hi = j-1;//下一次partition j 左边的subarray
else
return a[k];//j=k,返回
}
return a[k];//一直不满足,说明k>=N或k<=1,返回
}
算法细节:
- 对于已排序的数组,找最大的K个元素,上界是 NlogN N l o g N ,因为相当于二分法;下界是 N N <script type="math/tex" id="MathJax-Element-2">N</script>,相当于K很小的时候,起码要过一遍数组。