前端算法(52)

45 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

题目

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

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

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

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

题目解析

思路一

我们先定义三个指针:头指针start、尾指针end、移动指针cur并初始化,对于 nums[cur] === 2 交换 尾指针值和移动指针值,移动指针保持不变,尾指针 + 1。保证原尾指针的数据处理,对于 nums[cur] === 1 不做处理,移动指针直接跳过,cur + 1,对于 nums[cur] === 0 交换 头指针和移动指针值,并同时 +1,这样的可以保证了移动指针之前的所有数都不能为2, 遇到1跳过,遇到0在交换最前面的位置

/**
 * @param {number[]} nums
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var sortColors = function (nums) {
    let start = 0;
    let end = nums.length - 1;
    let cur = 0;

    const changeNums = (i, j) => {
        const temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }

    while (cur <= end) {
        const num = nums[cur];
        if (num === 2) {
            changeNums(cur, end)
            end--;
        } else if (num) {
            cur++
        } else {
            changeNums(cur, start)
            start++
            cur++
        }
    }

    return nums;
};

思路二

采用循环的方式进行解决,第一次循环遇到0就将前面的数字和当前指针进行交换,如此,循环一遍之后,所有的零就会在数组的最左边了,在进行第二次循环遇到1就将前面的数组和当前指针进行交换,如此,循环一遍之后, 所有的1也会到数组的左边了

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