leetcode 75 颜色分类

双指针法，分别指向下一个0的位置与下一个1的位置，值得注意的是，在遇到p0<p1，也就是p0当前位置指向一个1的时候，在i指向的数据与p0指向的数据交换后，需要将i指向的数据再次与p1指向的进行交换，因为第一步操作一定换出去了一个1。完成这一操作后，p0与p1都加1。贴代码

class Solution {
public:
    void sortColors(vector<int>& nums) 
    {
        int p_0 = 0;
        int p_1 = 0;
        int n = nums.size();
        for(int i = 0 ; i < n ; i++)
        {
            if(nums[i] == 1)
            {
                swap(nums[i],nums[p_1]);
                p_1++;
            }
            else if(nums[i] == 0)
            {
                swap(nums[i],nums[p_0]);
                if(p_0<p_1)
                swap(nums[i],nums[p_1]);
                p_0++;
                p_1++;
            }
        }
    }
};

还有一种双指针的办法，其实是一样的，殊途同归。

class Solution {
public:
    void sortColors(vector<int>& nums) 
    {
        int zero = 0;
        int two = nums.size();
        int i = 0;
        while(i<two)
        {
            if(nums[i] == 0)
            {
                swap(nums[i],nums[zero]);
                zero++;
                i++;
            }
            else if(nums[i] == 1)
            i++;
            else
            {
                two--;
                swap(nums[i],nums[two]);
            }
        }
    }
};