leetcode75.颜色分类

leetcode75.颜色分类

题目

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

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

用例

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

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

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

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

求解

/**
 * @param {number[]} nums
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var sortColors = function(nums) {
    let left = 0;
    let right = nums.length-1
    for(let i=0;i<=right;i++){
            if(nums[i]==0){
                let tmp = nums[left]
                nums[left] = nums[i]
                nums[i]=tmp
                left++
            }else if(nums[i]==2){
                let tmp = nums[right]
                nums[right]=nums[i]
                nums[i]=tmp
                right--
                i--
            }
    }
    return nums
};