前端算法专栏-数组-75.颜色分类

介绍

Hi 大家好。我是程序员库里，今天新开一个前端算法专栏。

接下来会分类给大家分享常考算法题目。

很多朋友也是看着这套系列算法拿到很多offer！所以也是想分享给更多朋友，帮助到有需要的朋友。

分类

数组-三路快排

题目

75. 颜色分类

给定一个包含红色、白色和蓝色、共 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]

解释

1.定义一个变量zero，初始值为-1，zero变量用来表示0…zero区间全部放0

2.定义一个变量two，初始值为数组的长度，two变量用来表示two…n-1区间全部放2

3.zero+1…n-1区间全部放1，这样数组中就变成了0…1…2

4.开始遍历数组，条件是当i小于数组长度的时候

5.如果遍历的当前元素是1，只把i向右移动一位，即i++。因为 1是在数组的中间，所以不做其他操作。

6.如果遍历的当前元素是2，先将变量two向左移动一位，腾出一个位置，也就是two–。然后将当前的元素2和two所在的位置交换一下位置，此时这个2就移动到了右边，这个时候不能将 i 向右移动一位，需要继续判断 当前这个元素是否为0

7.如果遍历的当前元素是0，先将zero向右移动一位，腾出一个位置，也就是zero++。然后将当前的元素0和zero所在的位置交换一下位置，此时这个0就移动到了左边。然后继续遍历，即i++。

8.遍历完一遍后，所有0就到了左边，所有2就到了右边，所有1就到了中间。即完成了数组排序。

代码

/**

* @param {number[]} nums

* @return {void} Do not return anything, modify nums in-place instead.

*/

var sortColors = function(nums) {

    let zero = -1;// [0...zero] 为0，弄成无效区间

    let two = nums.length;// [two...n-1] 为2，弄成无效区间

    for(let i =0;i<two;){

        if(nums[i] === 1){
            i++
        }else if(nums[i] === 2){
            two--
            [nums[i],nums[two]] = [nums[two],nums[i]]
        }else if(nums[i] === 0){
            zero++
            [nums[i],nums[zero]] = [nums[zero],nums[i]]
            i++
        }
    }
    return nums;
};