75.颜色分类

/**
 * @param {number[]} nums
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var sortColors = function (nums) {
    // 快速排序
    function quickSort(nums) {
        let x = arr.shift() // 分界值
        let leftArr = []
        let rightArr = []
        while (nums.length) {
            let n = nums.shift()
            if (n <= x) {
                leftArr.push(n)
            } else {
                rightArr.push(n)
            }
        }
        return quickSort(leftArr).concat(x, quickSort(rightArr))
    }

    // 插入排序
    function insertSort(nums) {
        // 假设第一个数是排好序的
        // 未排序的对已经排序的数从后往前扫，找合适位置插入
        for (let i = 1; i < nums.length; i++) {
            let cur = i
            for (let j = i - 1; j >= 0; j--) {
                if (nums[j] > nums[cur]) {
                    // 交换位置
                    let temp = nums[j]
                    nums[j] = nums[cur]
                    nums[cur] = temp
                    cur = j
                } else { // 插入
                    break
                }
            }
        }
        return nums
    }

    insertSort(nums)
    return nums
};

// 单指针
var sortColors = function (nums) {
    function swap(a, b) {
        let temp = a
        a = b
        b = temp
    }

    let p = 0
    // 分两次循环
    // 第一次循环把所有的0放到头部
    // 第二次循环把所有的1放到0的后面，那么2自然就会在尾部了
    for (let i = 0; i < nums.length; i++) {
        if (nums[i] === 0) {
            let temp = nums[i]
            nums[i] = nums[p]
            nums[p] = temp
            p++ // 向后扩充一个位置
        }
    }
    for (let i = p; i < nums.length; i++) {
        if (nums[i] === 1) {
            let temp = nums[i]
            nums[i] = nums[p]
            nums[p] = temp
            p++ // 向后扩充一个位置
        }
    }
    return nums
}

console.log(sortColors([2, 0, 2, 1, 1, 0]))