### 题目
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:
一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗?
### 思路
利用的双指针的办法,这样勉强算是一趟扫描吧。这个比排序算法简单的原因时因为数组内其实只有三种元素,最后要得到结果时把1放在中间,0放在最前面,2放在最后面,这样设置三个指针,模仿快排的思路,指针1代表下一个可以放置0的位置,指针2代表下一个可以放置2的位置,两个指针之间就是应该就是放1的位置,开始时将指针1指向0,指针2指向最后一个位置,指针3用来遍历数组,最开始指向第一个位置,指针3位于指针1和指针2之间。在遍历数组过程中,遇到1因为肯定在指针1和指针2之间,所以不管,遇到0,将其与指针1所指向的元素交换,然后指针1自加1,换回来的肯定是1,所以指针3也自加1,遇到2,和指针2所指的元素交换,因为换回来的元素不确定是等于多少,所以不能让指针3自加1,但是指针2要自减1,具体的看注释把。
### code
class Solution {
public:
void sortColors(vector<int>& nums) {
int p0=0;//指向放0的位置。
int p1=0;//用来遍历
int p2=nums.size()-1;//指向放2的位置
while(p1<=p2)//所有元素都遍历完了。
{
if(nums[p1]==0)swap(nums[p1++],nums[p0++]);//等于0,交换到最前面
else if(nums[p1]==2)swap(nums[p1],nums[p2--]);//等于2,交换到最后面。
else p1++;
}
}
};