【LeetCode 75】颜色分类

题目链接

【题解】

维护一个左边界l和一个右边界r
其中0..l-1都是'0' 而 r+1..n-1都是'2'
我们令i=l;i

【代码】

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