（js）leetcode 75. 颜色分类

秒速五厘米 2022-10-28 14:29 141阅读 0赞

### 题目： ###

> 给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
> 
> 此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
> 
> 示例 1：
> 
> 输入：nums = \[2,0,2,1,1,0\]  
> 输出：\[0,0,1,1,2,2\]  
> 示例 2：
> 
> 输入：nums = \[2,0,1\]  
> 输出：\[0,1,2\]  
> 示例 3：
> 
> 输入：nums = \[0\]  
> 输出：\[0\]  
> 示例 4：
> 
> 输入：nums = \[1\]  
> 输出：\[1\]
> 
> 提示：
> 
> n == nums.length  
> 1 <= n <= 300  
> nums\[i\] 为 0、1 或 2

### 思路： ###

> 使用双指针
> 
> [官方思路][Link 1]

### 代码实现： ###

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

### 运行结果： ###

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L01fRXZl_size_16_color_FFFFFF_t_70][]

[Link 1]: https://leetcode-cn.com/problems/sort-colors/solution/yan-se-fen-lei-by-leetcode-solution/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L01fRXZl_size_16_color_FFFFFF_t_70]: /images/20221024/8230d1928ba04224b555567b71987d76.png