Sort Colors

class Solution {
public:
    void sortColors(int A[], int n) {
        int num1,num2,num3,i;
        num1=num2=num3=0;
        for(i=0;i<n;i++)
        {
           if(A[i]==0) num1++;
           if(A[i]==1) num2++;
           if(A[i]==2) num3++;
        }
        int k=0;        
        while(num1--) A[k++]=0;
        while(num2--) A[k++]=1;
        while(num3--) A[k++]=2;
    }
};