本文已参与「新人创作礼」活动,一起开启掘金创作之路。
题目
给定一个包含红色、白色和蓝色、共 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
};