【Lintcode】颜色分类

给定一个包含红，白，蓝且长度为 n 的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。

我们可以使用整数 0，1 和 2 分别代表红，白，蓝。

class Solution {

    /**
     * @param nums: A list of integer which is 0, 1 or 2 
     * @return: nothing
     */
    public void sortColors(int[] nums) {
        // write your code here
        int a,b,c,n,i;
        a=0;b=0;c=0;
        n=nums.length;
        for(i=0;i<n;i++)
        {
            if(nums[i]==0)a=a+1;
            if(nums[i]==1)b=b+1;
            if(nums[i]==2)c=c+1;
        }
        for(i=0;i<a;i++)
        {
            nums[i]=0;
        }
        for(i=a;i<a+b;i++)
        {
            nums[i]=1;
        }
        for(i=a+b;i<a+b+c;i++)
        {
            nums[i]=2;
        }
        // nums[0]=n;
        // nums[1]=a;
        // nums[2]=b;
        // nums[3]=c;
    }
}