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

编程绿豆侠2025-06-13 11:01

这道题实际上就是让我们不用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++;
            }
        }
    }
};