LeetCode 75. 分类颜色

思路：定义3个下标，一个start和一个end,分别指向vector的前面和后面，一个可以活动的下标cursor，主要思想就是把遍历到的0和start指向的值交换，把遍历得到的2和end指向的值交换，这样就确保了vector的前面都是0，后面都是2，那么自然就排好序了。

class Solution {
public:
    void swap(int &a, int &b){
        int temp=a;
        a=b;
        b=temp;
    }
    
    void sortColors(vector& nums) {
        if(nums.size()==0 or nums.size()==1)
            return;
        
        int start=0, end=int(nums.size())-1, cursor=0;
        while(cursor<=end){
            if(nums[cursor]==0 && cursor==start){
                swap(nums[cursor], nums[start]);
                start++;
                cursor++; 
            }
            else if(nums[cursor]==0 && cursor!=start){
                swap(nums[cursor], nums[start]);
                start++;
            }
            else if(nums[cursor]==2){
                swap(nums[cursor], nums[end]);
                end--;
            }
            else
                cursor++;               
        }
    }
};