刷题-不同颜色排序

给定一个带有红色，白色或蓝色的n个对象的数组，对它们进行排序，使相同颜色的对象相邻，颜色顺序为红色，白色和蓝色。

在这里，我们将使用整数0,1和2分别表示红色，白色

class Solution {
public:
    void sortColors(int A[], int n) {

       int  l = 0, r = n - 1;
    for (int i = 0; i <= r; i++)
    {
        if (A[i] < 1)
        {
            A[i] = A[l]; A[l] = 0; l++;
        }
        else if (A[i] > 1)
        {
            A[i] = A[r]; A[r] = 2; r--;
            i --;//右边的换到左边与1未比较过
        }
    }
    }
};

三路快排的思想，以1作为标志，比1小的放在一边，比1大的放在另一边，但是要注意一点的是大的元素进行交换后由于该元素没有和标志1作比较，因此需要i=i-1