题目
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
示例
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
思路
此题可以使用双指针法解决,设置下标i,j,使得i所指的元素之前的位置均为0,j所指的元素之后的位置均为2,使得i,j之间的元素均为1。
代码:时间复杂度o(n),空间复杂度o(1)
void sortColors(vector<int>& nums) {
//i初始指向第一个元素,j指向最后一个元素
//k用来遍历
int i = 0, j = nums.size()-1, k = 0;
while(k <= j)
{
//如果k所指向的元素为0,就将其置换到最前面
if(nums[k] == 0){swap(nums[k],nums[i]);i++;k++;}
//如果k所指向的元素为2,就将其置换到最后面
else if(nums[k] == 2) {swap(nums[k],nums[j]); j--;}
//如果k所指向的元素为1,就继续向后遍历
else k++;
}
}