快速排序的关键主要就是partition函数,即将一个数组进行划分为左边的所有数都小于pivot,右边的所有的所有数都大于pivot,该partion函数有两种实现,双指针两边夹逼实现,还有一种就是双指针同向前进,本文主要采用双指针同向前进的方式。
快速排序的非递归实现主要就是用栈保存每一个待排序子串的首尾元素下标,下一次while循环时取出这个范围,对这段子序列进行partition操作。
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
//快速排序的非递归实现
//使用指针实现两个数的交换
void swap(int* a, int * b){
int temp =*a;
*a = *b;
*b = temp;
}
//使用引用实现两个数的交换
void swap2(int& a, int& b){
int temp = a;
b = a;
a = temp;
}
//使用双指针进行寻找pivot的最终放置位置, 递归版的和非递归版的partition都是可以的
int partition(vector<int>&arr, int low, int high){
int pivot = arr[low];
int l = low;//左指针表示目前小于等于pivot的最右边的位置
int r = low+1;
while(r<high){
if(arr[r]<pivot){
l++;
swap(arr[l], arr[r]);
}
r++;
}
//最终实现arr[l]和arr[low]的交换
swap(arr[low], arr[l]);
return l;
}
//使用递归进行求解
void quicksort1(vector<int>& arr, int low, int high){
if(low>=high) return;
int mid = partition(arr, low, high);
quicksort1(arr, low, mid);
quicksort1(arr, mid+1, high);
}
//使用非递归进行求解
//非递归主要就是借助栈进行保存每次的左右边界
void quicksort2(vector<int>& arr, int low, int high){
//申请一个栈
stack<int> s;
s.push(high);
s.push(low);
if(low<high){
while(!s.empty()){
int l = s.top();
s.pop();
int r = s.top();
s.pop();
int mid = partition(arr, l, r);
if(l<mid){//注意这个判断一定要加
s.push(mid);
s.push(l);
}
if((mid+1)<r){//注意这个判断一定要加
s.push(r);
s.push(mid+1);
}
}
}
}
int main(int argc, char *argv[]) {
vector<int> arr = {3, 2,-1,100, 0, 9, 8, 99, 200};
quicksort2(arr, 0, arr.size());
for(auto it:arr) cout<<it<<" ";
}
主要参考博文为:https://www.cnblogs.com/zhangchaoyang/articles/2234815.html