partition算法是快排的核心,还可用于TopK问题,它的思想是这样:从无序数组中随机选一个 pivot,然后通过一趟扫描,以 pivot 为分界线将数组中其他元素分为两部分,使得左边部分的数小于等于 pivot,右边部分的数大于等于 pivot(左部分或者右部分都可能为空),最后返回 pivot 在新的数组中的位置。
第一种写法:
public static int partition1(int[] a, int begin, int end) {
//选a[begin]作为pivot
int pivot = a[begin];
int pos = begin;
for (int i = begin + 1; i <= end; ++i) {
if (a[i] < pivot) {
pos++;
if (i != pos) //避免自身交换
swap(a, pos, i);
}
}
//最后交换pos和begin
swap(a, pos, begin);
return pos;
}
这种写法的思路是:选用 a[be