今天看到微博上有人转一篇文章,题目是《你问什么写不好快排》,看看自己多长时间能把快排写出来。结果,果然20分钟没写出来,本来以为明白原理,结果实现的出现各种问题,小 bug。最终花了半个点,写出来了,就是下面代码中 partition版本的。后来又翻了下算法导论,看到上面的快排,写了第二个分割子函数 partition2,这个原理更加清楚,用的是一个循环不变式的原理,将数组划分为了四个区域,具体说明可以参照算法导论。下面贴下代码,自己mark 一下。
算法导论上partition过程的伪代码如下
partition(A,p,r)
x <- A[r]
i <- p - 1
for j <- p to r - 1
do if A[j] <= x
then i <- i +1
exchange A[i] <-> A[j]
exchange A[i+1] <-> A[r]
return i+1
1) 如果 p <= k <= i,则 A[k] <= x
2) 如果 i + 1 <= k <= j - 1,则A[k] > x
3) 如果k = r,则 A[k] = x
package com.deron.business;
/**
* this is quick sort tool class
*
* @author deron
*
*/
public class QSort {
public static void qSort(int[] array) {
qSort(array, 0, array.length - 1);
}
public static void qSort(int[] array, int start, int end) {
if (start < end) {
int t = partition2(array, start, end);
qSort(array, start, t - 1);
qSort(array, t + 1, end);
}
}
public static int partition(int[] array, int start, int end) {
int t = start;
int lp = start + 1;
int rp = end;
while (lp < rp) {
while (array[lp] <= array[t] && lp < rp)
lp++;
while (array[rp] >= array[t] && lp < rp)
rp--;
if (lp < rp) {
int tmp = array[lp];
array[lp] = array[rp];
array[rp] = tmp;
lp++;
rp--;
}
}
if (array[lp] <= array[t]) {
int tmp = array[t];
array[t] = array[lp];
array[lp] = tmp;
t = lp;
} else {
int tmp = array[t];
array[t] = array[lp - 1];
array[lp - 1] = tmp;
t = lp - 1;
}
return t;
}
public static int partition2(int[] array, int start, int end) {
int v = array[start];
int lp = start;
int rp = start + 1;
for (; rp <= end; rp++) {
if (array[rp] <= v) {
lp++;
int tmp = array[lp];
array[lp] = array[rp];
array[rp]= tmp;
}
}
array[start] = array[lp];
array[lp]= v;
return lp;
}
}