题解 | #排列颜色#

题目描述：现在有一个包含n个物体的数组，其中物体颜色为颜色为红色、白色或蓝色，请对这个数组进行排序，让相同颜色的物体相邻，颜色的顺序为红色，白色，蓝色。我们用0,1,2分别代表颜色红，白，蓝
注意：本题要求你不能使用排序库函数
思路：一个非常直接的解法是两步的计数排序的算法
                1、首先：遍历一遍数组，记录0,1,2的数量；
                2、然后重写这个数组，先将0写入，再将1写入，再将2写入

class Solution {
public:
    void sortColors(int A[], int n) 
    {
        int M=0,N=0,L=0;
        for(int i=0;i<n;i++)
        {
            if(A[i] == 0)
                M++;
            else if(A[i] == 1)
                N++;
            else if(A[i] == 2)
                L++;
        }
        for(int i=0;i<n;i++)
        {
            if(i<M)
                A[i] = 0;
            else if(i <(M+N))
                A[i]=1;
            else
                A[i] =2;
        }
    }
};