每日一道算法题--leetcode 75--颜色分类（快排）--C++-优快云博客

本文介绍了一种高效的数组排序方法，特别适用于三色问题。通过两种不同的实现方式——计数排序和快速排序来解决这一问题，并详细解释了每种方法的具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目描述】

【代码思路】

最简单的思路是扫描一遍数组分别对三个类别的数量做个统计，然后再按照每类的数量给数组赋值即可。代码如下：

class Solution {
public:
    void sortColors(int A[], int n) {
        int i = 0;
        int j = 0;
        int k = 0;
        for(int p = 0; p < n; p ++)
        {
            if(A[p] == 0)
            {
                i ++;
            }
            else if(A[p] == 1)
            {
                j ++;
            }
            else
                k ++;
        }

        for(int p = 0; p < n; p ++)
        {
            if(p < i)
                A[p] = 0;
            else if(p >= i && p < i + j)
                A[p] = 1;
            else
                A[p] = 2;
        }
    }
};
复制代码

换个思路的话，这道题其实就是排序，我想到的是快排，是排序中的基础方法,顺便就练习一下。把数组中的第一个元素作为哨，我们知道一趟快排结束之后，哨将找到它最终排序所在的位置，再递归的对哨左边和右边进行快排即可。简单介绍一下快排，图片来自百度百科：

【源代码】 快排的源代码

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int left=0;
        int right=nums.size()-1;
        Qsort(nums,left,right);
    }
    void Qsort(vector<int>& a, int low, int high){
    if(low >= high)
    {
        return;
    }
    int first = low;
    int last = high;
    int key = a[first];
 
    while(first < last)
    {
        while(first < last && a[last] >= key)
        {
            --last;
        }
 
        a[first] = a[last];
 
        while(first < last && a[first] <= key)
        {
            ++first;
        }
         
        a[last] = a[first];    
    }
    a[first] = key;
    Qsort(a, low, first-1);
    Qsort(a, first+1, high);
}
};
复制代码