面试必备: 三路快速排序

xhchen2024

已于 2023-07-19 23:41:47 修改

阅读量53

点赞数 1

分类专栏：数据结构与算法文章标签：面试算法快速排序排序

于 2023-07-19 23:39:34 首次发布

本文链接：https://blog.csdn.net/weixin_40519680/article/details/131819922

版权

数据结构与算法专栏收录该内容

3 篇文章 0 订阅

订阅专栏

思路

快速排序是一种平均时间复杂度 $O(n\text{log}n)$ 的高效排序算法, 但在一些特殊情况(例如数组中有大量重复元素)下时间复杂度容易退化成 $O(n^2)$ , 一种解决方案是使用三路快速排序, 其思路如下: 设排序区间为 $[s, e)$ , 首先在区间内随机选一个基准 $p i v o t$ , 接下来的目标是将区间划分为三个部分: 1) $n u m s [k] < p i v o t$ , 2) $n u m s [k] == p i v o t$ , 3) $n u m s [k] > p i v o t$ . 具体通过在遍历数组的过程中维护两个指针 $l, r$ 使得: $[s, l)$ 区间内的数 $\; <pivot$ 且 $[r, e)$ 区间内的数 $\; >pivot$ . 划分完后对 $[s, l)$ 区间和 $[r, e)$ 区间递归调用排序函数.

实现代码

下面是一道用三路快排的模板题代码:排序数组, 因为题目测试数据补了一些“特殊数据”所以之前的一般快排就过不了了, 要考虑那些corner case就可以用三路快排来做. 当然如果只是为了过这道题用归并排序或堆排序等稳定 $O(n\text{log}n)$ 的排序算法也是可以的~

class Solution {
public:
    void qsort(vector<int> &li, int s, int e) {//三路快排
        if (e - s <= 1)
            return;
        int pivot = li[rand() % (e - s) + s];
        int l = s, r = e, i = s;
        while (i < r) {
            if (li[i] < pivot)
                swap(li[l++], li[i++]);
            else if (li[i] > pivot)
                swap(li[i], li[--r]);
            else
                i++;
        }
        qsort(li, s, l);
        qsort(li, r, e);
    }

    vector<int> sortArray(vector<int> &nums) {
        qsort(nums, 0, nums.size());
        return nums;
    }
};