给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地 对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。
示例 1:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
可以用左右两个指针和一个 cur 指针。cur 本来指向最左边,当发现 0 时和左指针指向的数交换;当发现 1 时,向右移动;当发现 1 时和右指针指向的数交换。
注意:当和左边交换时,我们可以确定左边交换过来的一定是 1,因此交换之后 cur 可以跳到下一个,而和右边交换完之后则不能跳过当前值,应该进入下一轮循环。
C++15 linesExpandCollapse
void sortColors(vector<int>& nums) {
int l = 0, cur = 0, r = nums.size() - 1;
while (cur <= r) {
if (nums[cur] == 0) {
swap(nums[cur], nums[l]);
l++;
cur++;
} else if (nums[cur] == 1) {
cur++;
} else {
swap(nums[cur], nums[right]);
right--;
}
}
}