LeetCode 热题 HOT 100之颜色分类

方案：使用双指针，p0 和p1代表数字0和数字1的位置，遍历数组遇到0时，与p0指向位置进行交换，两个指针后移，遇到1时1，与p1指针位置交换，p1后移。

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

来源：力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。