Simplest solution counting the numbers c++


#1

void Solution::sortColors(vector &A) {

int n=A.size();
int a=0,b=0,c=0;

for(int i=0;i<n;i++){
    if(A[i]==0){
        a++;
    }
    if(A[i]==1){
        b++;
    }
    if(A[i]==2){
        c++;
    }
}

for(int i=0;i<n;i++){
    if(i<a){
        A[i]=0;    
    }
    
    if(i<a+b && i>=a){
        A[i]=1;    
    }
    if(i<a+b+c && i>=a+b){
        A[i]=2;    
    }
}
return;

}