题目描述
给定一个包含红色、白色和蓝色共 n个元素的数组nums,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数0、1和2分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。
示例 1:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
示例 2:
输入:nums = [2,0,1]
输出:[0,1,2]
你能想出一个仅使用常数空间的一趟扫描算法吗?
解题思路
简单:排序
(0.0不赘述了)
这里的数据只有0、1、2三种,可以使用更加特殊的方法做。
简单:2次遍历
第一次遍历:用3个int,统计3种颜色的个数,假设有红色x个、白色y个和蓝色z个
第二次遍历:赋值。
0 ~ (x-1) 红色
x ~ (x+y-1)白色
x+y ~ n-1 蓝色
//第1次遍历:用3个int,统计3种颜色的个数
int cnt[3]={0};
for(int i=0;i<nums.size();i++){
cnt[nums[i]]++;
}
//第2次遍历
for(int i=0;i<nums.size();i++){
if(i<x) nums[i]=0;
else if(i<x+y) nums[i]=1;
else nums[i]=2;
}
进阶:1次遍历
0.一开始思路是用双指针,一个从前往后负责识别0,一个从后往前负责识别2。对比前大于后,则交换
但这样是错误的!原因:中间1的区间中指针没办法推进。
所以多引进一个指针来负责1的区域,就是下面的3指针方法。
1.三指针(题解中的双指针)
分别用三个指针标记三个颜色的区域
- zero作为标记0与1的分界线
- two标记1与2的分界线
- i负责遍历1的区间将区间内的0、2交换出去。
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
void sortColors(vector<int> &nums) {
int size = nums.size();
if (size < 2) {
return;
}
// all in [0, zero) = 0
// all in [zero, i) = 1
// all in [two, len - 1] = 2
// 循环终止条件是 i == two,那么循环可以继续的条件是 i < two
// 为了保证初始化的时候 [0, zero) 为空,设置 zero = 0,
// 所以下面遍历到 0 的时候,先交换,再加
int zero = 0;
// 为了保证初始化的时候 [two, len - 1] 为空,设置 two = len
// 所以下面遍历到 2 的时候,先减,再交换
int two = size;
int i = 0;
// 当 i == two 上面的三个子区间正好覆盖了全部数组
// 因此,循环可以继续的条件是 i < two
while (i < two) {//1的区间还没有遍历完
if (nums[i] == 0) {
swap(nums[zero], nums[i]);
zero++;
i++;
} else if (nums[i] == 1) {
i++;
} else {
two--;
swap(nums[i], nums[two]);
}
}
}
};
/*
作者:liweiwei1419
链接:https://leetcode.cn/problems/sort-colors/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
*/
2.巧妙赋值
这个是题解中一个很巧妙的赋值方法!
题解链接:最简洁的双指针解法!一次遍历,无交换操作
大致思路:遍历的同时统计0和1的个数,并将数组元素首先全部赋为2。在统计过程中,每遇到1,则前n1个元素必为0或1;每当遇到0,前n0个元素必为0。于是,每当n1++,前n1个值刷成1,每当n0++,前n0个值覆盖掉之前的1赋为0。
我对代码的解读放在注释中
public void sortColors(int[] nums) {
//n0记录0个个数, n1记录0和1的个数
int n0 = 0, n1 = 0;
//遍历数组
for(int i = 0; i < nums.length; i++){
int num = nums[i];
nums[i] = 2; //首先将数组中所有元素赋值2
//统计<2的个数,并将前n1个都赋为1
if(num < 2){
nums[n1++] = 1;
}
//统计=0的个数,并将前n0个都赋为0
//由于n0<=n1,n0会将n1更新过的值覆盖掉
if(num < 1){
nums[n0++] = 0;
}
}
}
/*
作者:颠倒的蝴蝶
链接:https://leetcode.cn/problems/sort-colors/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
*/