高级排序算法-快速排序
这个方法的代码参考leecode平台的大佬——liweiwei1419的文章,原文出处:https://leetcode.cn/leetbook/read/learning-algorithms-with-leetcode/55szu2/
本文语言用的是C++:
快速排序实现一(大放过,小操作):
//leecode第912题快速排序一实现
class Solution {
public:
vector<int> sortArray(vector<int>& nums) {
int len = nums.size();
quickSort(nums, 0, len - 1);
return nums;
}
private:
void quickSort(vector<int>& nums, int left, int right) {
// 注意:这里包括 > 的情况,与归并排序不同,请通过调试理解这件事情
if (left >= right) {
return;
}
int p = partition(nums, left, right);
quickSort(nums, left, p - 1);
quickSort(nums, p + 1, right);
}
//最基础的快速排序,大方过,小操作思想,循环不变量:
// [left + 1, lt] < pivot,
// [lt + 1, i) >= pivot
private:
int partition(vector<int>& nums, int left, int right) {
// 随机选择一个元素作为切分元素
int randomIndex = rand() % (right - left + 1) + left;//生成随机数的范围[left~right]
swap(nums, randomIndex, left);
int pivot = nums[left];
int lt = left;
// 注意,这里取等号
for (int i = left + 1; i <= right; i++) {
if (nums[i] < pivot) {
// 交换当前元素与 lt 的位置
lt++;
swap(nums, i, lt);
}
}
// 最后这一步要记得交换到起始位置
swap(nums, left, lt);
return lt;
}
private:
void swap(vector<int>& nums, int index1, int index2) {
int temp = nums[index1];
nums[index1] = nums[index2];
nums[index2] = temp;
}
};
快速排序实现二(指针对撞):
class Solution {
public:
vector<int> sortArray(vector<int>& nums) {
int len = nums.size();
quickSort(nums, 0, len - 1);
return nums;
}
private:
void quickSort(vector<int>& nums, int left, int right) {
// 注意:这里包括 > 的情况,与归并排序不同,请通过调试理解这件事情
if (left >= right) {
return;
}
int p = partition(nums, left, right);
quickSort(nums, left, p - 1);
quickSort(nums, p + 1, right);
}
// 快速排序2,:在lg,ge两指针对撞过程中排序,循环不变量:
// [left + 1, le) <= pivot
// (ge, right] >= pivot
int partition(vector<int>& nums, int left, int right) {
// 随机选择一个元素作为切分元素
int randomIndex = rand() % (right - left + 1) + left;//生成随机数的范围[left~right]
swap(nums, randomIndex, left);
int pivot = nums[left];
int le = left + 1;
int ge = right;
while (true) {
// 注意:这里一定是 nums[le] < pivot,等于 pivot 的元素是被交换过来得到的
while (le <= ge && nums[le] < pivot) {
le++;
}
// 此时 le 来到第 1 个大于等于 pivot 的位置
while (le <= ge && nums[ge] > pivot) {
ge--;
}
// 此时 ge 来到第 1 个小于等于 pivot 的位置
if (le > ge) {
break;
}
swap(nums, le, ge);
le++;
ge--;
}
swap(nums, left, ge);
return ge;
}
void swap(vector<int>& nums, int index1, int index2) {
int temp = nums[index1];
nums[index1] = nums[index2];
nums[index2] = temp;
}
};
快速排序实现三(三向切分):
理解代码的关键是把握好循环不变量
class Solution {
public:
vector<int> sortArray(vector<int>& nums) {
int len = nums.size();
quickSort(nums, 0, len - 1);
return nums;
}
private:
//初始时设定lt,gt,i三个指针,lt,i指向最左侧,gt指向
//最右侧,然后随着i向右遍历,同时受循环不变量的约束,lt,
//gt向中间靠拢,当i和gt碰撞,则退出循环,此时整个区间
//就被lt,gt切分为三个部分,并且满足循环不变量的限制,遍历过程循环不变量:
//[left + 1, lt] < pivot
//[lt + 1, i) == pivot
//[gt, right] > pivot
void quickSort(vector<int>& nums, int left, int right) {
if(left >= right){
return;
}
int randomIndex = rand() % (right - left + 1) + left;//生成随机数的范围[left~right]
swap(nums, randomIndex, left);
int pivot = nums[left];
//为了使循环不变量的初始区间为空,应该进行下面这些初始化,即初始化为[a-1,a],[a,a),(a,a]的形式
int lt = left;
int gt = right + 1;
int i = lt + 1;
while(i < gt){
if(nums[i] < pivot){
lt++;
swap(nums, i, lt);
i++;
}
else if(nums[i] == pivot){
i++;
}
else if(nums[i] > pivot){
gt--;
swap(nums, i, gt);
}
}
//经循环不变量之后,跳出循环的情况应该(i==gt),此时区间满足:
//[left + 1, lt] < pivot
//[lt + 1, gt) == pivot , 即[lt + 1, gt - 1] == pivot
//[gt, right] > pivot
//因此我们交换pivot元素到lt的位置即是其最终的排序位置
swap(nums, left, lt);
//此时三个区间有如下关系:
//[left , lt-1] < pivot
//[lt, gt) == pivot , 即[lt + 1, gt - 1] == pivot
//[gt, right] > pivot
//很显然,此时[lt, gt)区间的所有元素由于都等于pivot,因此它们所处位置均是最终排序位置,接下来递归处理剩下
//未定的区间即可,这是分治思想的体现!
quickSort(nums, left, lt - 1);
quickSort(nums,gt,right);
}
private:
void swap(vector<int>& nums, int index1, int index2) {
int temp = nums[index1];
nums[index1] = nums[index2];
nums[index2] = temp;
}
};