leetcode 75 颜色分类 (c++)

### 题目

给定一个包含红色、白色和蓝色，一共 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++;
        }
    }
};