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

2023-11-29 09:34:48 浏览数 (1)

介绍

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

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

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

分类

数组-三路快排

题目

75. 颜色分类

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

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

必须在不使用库内置的 sort 函数的情况下解决这个问题。

示例 1:

代码语言:txt复制
输入: nums = [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

示例 2:

代码语言:txt复制
输入: 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就到了中间。即完成了数组排序。

代码

代码语言:javascript复制
/**

* @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;
};

0 人点赞