力扣HOT100之技巧:75. 颜色分类

这道题实际上就是让我们不用sort()函数来实现对原数组的排序,这里我直接使用快速排序对原数组进行排序了,也是复习一下基于快慢指针的快速排序写法。面试手撕快排的思路参考这个视频

用时击败100%,还行。下面直接贴代码。

cpp 复制代码
class Solution {
public:
    void sortColors(vector<int>& nums) {
        quicksort(nums, 0, nums.size() - 1);
    }
    void quicksort(vector<int>& nums, int begin, int end){
        if(begin >= end) return ;   //只有一个元素,无需排序,直接返回
        //排序主逻辑
        int pivot = nums[end];
        int slow = begin, fast = begin; 
        while(fast < end){
            if(nums[fast] > pivot)
                fast++;
            else{
                swap(nums[slow], nums[fast]);
                slow++;
                fast++;
            }
        }
        swap(nums[slow], nums[end]);  //将枢轴移动到正确位置
        quicksort(nums, begin, slow - 1);   //对枢轴左侧排序
        quicksort(nums, slow + 1, end);   //对枢轴右侧排序
    }
};

但是,使用快速排序无法通过一次扫描就将整个数组排序完成,还可以使用更高效的方法。我感觉灵神的题解也写得挺通俗易懂的,感觉灵神的题解写的特别好,还是建议去看他的题解

以下是根据灵神的题解写出来的代码。

cpp 复制代码
class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p0 = 0, p1 = 0;
        for(int i = 0; i < nums.size(); i++){
            int temp = nums[i];
            nums[i] = 2;   
            if(temp <= 1){
                nums[p1] = 1;
                p1++;
            }
            if(temp == 0){
                nums[p0] = 0;
                p0++;
            }
        }
    }
};
相关推荐
pumpkin845142 分钟前
Rust 调用 C 函数的 FFI
c语言·算法·rust
挺菜的14 分钟前
【算法刷题记录(简单题)003】统计大写字母个数(java代码实现)
java·数据结构·算法
mit6.82414 分钟前
7.6 优先队列| dijkstra | hash | rust
算法
2401_858286111 小时前
125.【C语言】数据结构之归并排序递归解法
c语言·开发语言·数据结构·算法·排序算法·归并排序
guygg881 小时前
基于matlab的FIR滤波器
开发语言·算法·matlab
ysh98882 小时前
PP-OCR:一款实用的超轻量级OCR系统
算法
遇雪长安2 小时前
差分定位技术:原理、分类与应用场景
算法·分类·数据挖掘·rtk·差分定位
数通Dinner2 小时前
RSTP 拓扑收敛机制
网络·网络协议·tcp/ip·算法·信息与通信
张人玉4 小时前
C# 常量与变量
java·算法·c#
weixin_446122465 小时前
LinkedList剖析
算法