快速排序的实现分为两个步骤:
1. 将输入的数组取pivot,按照各个数据的大小关系,移动`为左右两段(< pivot及>= pivot)
2. 对左段和右段递归处理。
递归好说,不好记的在于如何对于一个数组,取pivot之后将其他数据按大于小于关系,移动到左右,构成递归下一层的两段。
这里的实现是,用i和j分别从前后向中间移动,i指向的是从前往后的第一个>=pivot的值;j指向的是从后往前的第一个< pivot的值,两者交换,完成一次循环(第一层while)。
容易忽略但很必要的是,在循环中,每个地方都要检查i < j。
int AdjustArray(int arr[], int start, int end)
{
int i = start, j = end;
const int pivot = arr[start]; // 取第一个作为pivot,记住之后该位置可被覆盖
// 每次循环,都会尝试前后交换一对数值
while (i < j)
{
// 移动j,从后往前找第一个小于pivot的数,往前面填
while (i < j && arr[j] >= pivot)
j--;
if (i < j)
{
arr[i] = arr[j];
i++;
}
// 此时j指向的位置空出,等待一个>=pivot的数来填
// 移动i,从前往后找第一个大于pivot的数(应处于最终pivot右),往j所在的位置填
while(i < j && arr[i] <= pivot)
i++;
if(i < j)
{
arr[j] = arr[i];
j--;
}
// 此时i指向一个可被填的空位,为下一轮循环可用
// 循环中全程都要判断i < j!
}
arr[i] = pivot;
return i;
}
void QuickSort(int arr[], int start, int end)
{
if (start < end)
{
// divide
int mid = AdjustArray(arr, start, end);
// conquer
QuickSort(arr, start, mid - 1);
QuickSort(arr, mid + 1, end);
}
}