颜色分类算法

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

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

必须在不使用库的sort函数的情况下解决这个问题。

示例 1：

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

示例 2：

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

提示：

• n == nums.length
• 1 <= n <= 300
• nums[i] 为 0、1 或 2

快排思想

使用三指针，维护两个区（0区和2区），遍历数组把元素划分到对应区中

var sortColors = function(nums) {
    if(!nums.length) return []
    const swap = (a,b) => {
        let temp = nums[a]
        nums[a] = nums[b]
        nums[b] = temp
    }
    let [cur,left,right] = [0,0,nums.length - 1]
    while(cur <= right){
        if(nums[cur] == 0){
            // 当前数为0，就往0区放，left++，cur++
            // cur不可能在left左边
            swap(left,cur)
            left++
            cur++
        }else if(nums[cur] == 2){
            // 当前数为2，就往2区放，right--
            // 此处不能cur++,因为交换过来的可能为0，还需判断
            swap(cur,right)
            right--
        }else {
            cur++
        }
    }
    return nums
};

题目来源-力扣，如有纰漏欢迎指正