LeetCode 75 颜色分类

颜色分类

class Solution {
public:
    void sortColors(vector<int>& nums) {
        //使用三路快速排序算法
        int len=nums.size();
        int zeros=-1;
        int one=0;//其实就是归并的标识
        int two=len;
        for(;one<two;)
        {
            if(nums[one]==0)
            {
                ++zeros;
                swap(nums[zeros],nums[one]);
                ++one;
            }
            else if(nums[one]==2)
            {
                --two;
                swap(nums[two],nums[one]);              
            } 
            else
                ++one;
        }  
    }
};