快排非递归算法

时间:2021-02-02 04:13:33
//如有错误,望指正
#include <iostream>
#include <stack>
#include <assert.h>
//非递归快排

using namespace std;

void create(int a[], int n)
{
for(int i=0; i<n ;i++)
a[i] = rand()%(10*n);
}

void print(int a[], int l,int r)
{
for(int i=l; i<=r; i++)
cout<<a[i]<<" ";
cout<<endl;
}

void myassert(int a[], int n)
{
for(int i=1; i<n; i++)
assert(a[i]>=a[i-1]);
}

void qsort(int a[], int n)
{
stack<int> st;
st.push(n);
st.push(0);
while(st.size()>1)
{
int l = st.top();//作为左边界
st.pop();
int r = st.top()-1;//st.top()-1作为右边界,而st.top()下一个子数组的右边界,所以不要pop出来
//print(a,l,r);

if(l>=r)
continue;
int mid = (a[l]+a[r])/2;
int i = l, j = r;
while(i<j)
{
while(i<j && a[i]<=mid)i++;
while(i<j && a[j]>mid)j--;//不能是>=,若为>=,则会使2,1这样的例子不能交换;
if(i<j)swap(a[i],a[j]);
}
assert(i==j);
assert(a[i]>=mid);

st.push(j);
st.push(l);
}
st.pop();
}

int main()
{
const int SIZE = 100;
int* a = new int[SIZE];

create(a,SIZE);

print(a,0,SIZE-1);
qsort(a,SIZE);
print(a,0,SIZE-1);

myassert(a,SIZE);

delete[] a;

return 0;
}