[路飞]_LeetCode_75. 颜色分类

「这是我参与2022首次更文挑战的第6天，活动详情查看：2022首次更文挑战」

题目

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

来源：力扣（LeetCode）leetcode-cn.com/problems/so…

解题思路

题目描述中有说 原地对它们进行排序，使得相同颜色的元素相邻，那么我们把数组进行排序就可以了，排序有很多种方法，任选一种都可以，这里我们用快速排序。另外因为数组中只存在 0、1、2 三种元素，所以使用计数排序也是个不错的选择，元素少是计数排序最好的使用场景。

最后注意，题目要求我们在原数组的基础上进行排序，不需要返回任何结果。

代码实现

var sortColors = function(nums) {
    quikSort(nums, 0, nums.length - 1)
};

var quikSort = function(nums, start, end) {
    //快速排序是用分治的思想，把数组分为大小两个区，再分别对两块区再进行快速排序
    if(start < end) {
        // 每次确定一位数字的位置
        let base = partition(nums, start, end)
        quikSort(nums, start, base - 1)
        quikSort(nums, base + 1, end)
    }
}

var partition = function(nums, start, end) {
    //从待排序区找一个基准数，把小于这个数的数放左边，大于这个数的数放右边
    //每一轮对比完成后，基准数的位置确定下来
    //再递归对左边、右边部分做相同操作

    let base = end //选最后一位为基准数的位置
    let bigStart = start //默认开始位置为大数的开始位置

    for (let i = start; i < end; i++) {
        //当前数小于基数时，把当前数和第一个大数交换位置
        if (nums[i] <= nums[base]) {
            [nums[bigStart], nums[i]] = [nums[i], nums[bigStart]]
            minEnd++
        }
    }
    
    //基数和第一个大数交换位置
    [nums[bigStart], nums[base]] = [nums[base], nums[bigStart]]

    //返回当次确定位置的下标
    return minEnd
}

如有错误欢迎指出，欢迎一起讨论！