算法思想
- 如果要排序数组中下标从 p 到 r 之间的一组数据,我们选择 p 到 r 之间的任意一个数据作为 pivot(分区点)。我们遍历 p 到 r 之间的数据,将小于 pivot 的放到左边,将大于 pivot 的放到右边,将 pivot 放到中间。经过这一步骤之后,数组 p 到 r 之间的数据就被分成了三个部分,前面 p 到 q-1 之间都是小于 pivot 的,中间是 pivot,后面的 q+1 到 r 之间是大于 pivot 的。根据分治、递归的处理思想,我们可以用递归排序下标从 p 到 q-1 之间的数据和下标从 q+1 到 r 之间的数据,直到区间缩小为 1,就说明所有的数据都有序了。
代码实现
public void partition(int[] list, int startIdx, int endIdx) {
if(startIdx == endIdx - 1) {
if(sortType == SortType.Positive.type() ? list[startIdx] > list[endIdx] : list[startIdx] < list[endIdx]) {
int tmp = list[startIdx];
list[startIdx] = list[endIdx];
list[endIdx] = tmp;
}
return;
}
int flag = startIdx;
int pivot = list[endIdx];
for(int k = startIdx; k < endIdx ; k++) {
if(sortType == SortType.Positive.type() ? list[k] < pivot : list[k] > pivot) {
int tmp = list[flag];
list[flag] = list[k];
list[k] = tmp;
flag++;
}
}
// 如果游标不是最后一位,游标位置和最后的标志位互换
if(flag < endIdx) {
int tmp = list[flag];
list[flag] = list[endIdx];
list[endIdx] = tmp;
}
// 处理小的区间
// 如果小区间的元素大于1个
if(startIdx + 1 < flag) {
partition(list, startIdx, flag - 1);
}
// 处理大的区间
// 如果大区间一个元素都没有了
if(flag + 1 < endIdx) {
partition(list, flag + 1, endIdx);
}
}
@Override
public void testSort(int[] list) {
partition(list, 0, list.length - 1);
}
完整代码见:快速排序