描述
1)基于递归的排序容易造成栈溢出,因此这里提供一种快排的非递归版本;
2)由于使用到了stack数据结构以存储需要快排的数据段,因此也存在内存溢出的风险;
3)如果希望避免内存溢出,可以考虑混合式排序,即在STL中的sort函数,是一种混合排序(自省式排序)
——如果递归深度过深,采用堆排序;如果当前排序元素过少,则使用插入排序(插入排序在数组相对有序的情况下,性能比快排好);否则是使用三数取中快排;
源码
//三数取中,将中值放在first上
void swapThreeMedian(int& first, int& second, int& third) {
if (second < first) std::swap(second, first);
if (second < third) std::swap(second, third);
if (first < third) std::swap(first, third);
}
//三数取中快排(非递归版本)
void quickSort2(std::vector<int>& nums, int begin, int end) {
if (end - begin <= 1) return;
std::stack<std::pair<int, int>>stack;
stack.push(std::make_pair(begin, end));
while (!stack.empty()) {
auto p = stack.top();
stack.pop();
int begin = p.first, end = p.second;
if (end - begin <= 1) continue;
swapThreeMedian(nums[begin], nums[begin + (end - begin) / 2], nums[end - 1]);
int left = begin + 1;
int right = end - 1;
int pivot = begin;
while (left <= right) {
while (left <= right && nums[left] <= nums[pivot]) ++left;
while (left <= right && nums[right] > nums[pivot])--right;
if (left < right)
std::swap(nums[left], nums[right]);
}
std::swap(nums[pivot], nums[right]);
stack.push(std::make_pair(begin, right));
stack.push(std::make_pair(right+1, end));
}
}