（每日一练java）CC74 排列颜色

描述

现在有一个包含n个物体的数组，其中物体颜色为颜色为红色、白色或蓝色，请对这个数组进行排序，让相同颜色的物体相邻，颜色的顺序为红色，白色，蓝色。

我们用0,1,2分别代表颜色红，白，蓝

注意：

本题要求你不能使用排序库函数

扩展：

一个非常直接的解法是两步的计数排序的算法

首先：遍历一遍数组，记录0,1,2的数量，然后重写这个数组，先将0写入，再将1写入，再将2写入

你能给出一个只用一步，并且能在常数级空间复杂度解决这个问题的算法吗？

class Solution {
public:
    void myswap(int &a, int &b){
        int t = a;
        a = b;
        b = t;
	}
    void sortColors(int A[], int n) {
        int l = 0;
        int r = 0;
        for (int i = 0; i < n-r; i++){
            if (A[i] == 0){
                myswap(A[i], A[l]);
                l++;
            }
            else if(A[i]==2){
                myswap(A[i], A[n - 1 - r]);
                r++;
                i--;
            }
        }
    }
};