基于快排的思想

回忆一下快排的思路:

把第一个位置的作为pivot,从右遍历直到发现有比pivot小的,交换;从左遍历,直到发现有比pivot大的,交换

注意这里交换的都是当前的low和high,直到low,high相遇,本轮停止

如果数字里面只有0,1,2那么显然只需要一次while就可以结束

思路依旧是l r current

current==0 需要和low交换,使得0都在左侧

current==1 没问题,毕竟现在左边的都是0,右边的都是2。继续遍历

current==2 需要和high交换,使得2都在右侧

归根结底还是对于快排的思想掌握的不好

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