leetcode 75. 颜色分类

思路一：
0换到前面，2换到后面

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int i=0,j=nums.size()-1;
        int temp;
        for(int k=0;k<nums.size();k++){
            if(nums[k]==0){
                swap(nums[k],nums[i]);
                i++;
            }
            else if(nums[k]==2){
                if(k<j) {
                    swap(nums[k],nums[j]);
                    j--;
                    k--;//否则后面交换到前面的数字并没有被梳理（比较然后安插）到正确位置
                }
            }
        }

// 输入：
// [1,2,0]
// 输出：
// [1,0,2]
// 预期结果：
// [0,1,2]
        
    }
};

思路二：
0和1设指针

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