LeetCode 75. 颜色分类 javascript实现

题目

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

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

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶：

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-colors
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

这道题的解题其实就只是把0放在最前面，把2放在最后面，把1放在中间，其实一个排序就可以完成了，但是因为数组中只有三个值，所以我们可以使用三指针的方式，用O(n)的时间复杂度就完成了
三个指针分别指向

• 当前已知0放置在最右边的下标 left
• 当前已知1放置在最左边的下标 right
• 当前正在遍历的下标 index
  遍历的时候有三种情况

1. 当遍历到的数为0的时候，判断当前的index和left是否相同，如果相同，就直接将两个下标自增，如果不同，就将两个下标对应的值交换，然后left自增
2. 当遍历到的数为1的时候，index自增
3. 当遍历到的数为2的时候，index和right两个下标对应的数值交换，然后right自减

结束条件

1. left>=right
2. index>right

实现代码

var sortColors = function(nums) {
    let left = 0;
    let right = nums.length-1;
    let index = 0;
    let swap;
    while(left<right&&index<=right){
        if(nums[index]===0){
            if(index!==left){
                swap = nums[left];
                nums[left] = nums[index];
                nums[index] = swap;
            }else{
                index++;
            }
            left++;
        }else if(nums[index] === 1){
            index++;
        }else{
            swap = nums[right];
            nums[right] = nums[index];
            nums[index] = swap;
            right--;
        }
    }
    return nums;
};