Easy solution using count method


#1

void Solution::sortColors(vector &A) {
int a=0,b=0,c=0;
for(int i=0;i<A.size();i++){
A[i]==0? a++:A[i]==1? b++:c++;
}
for(int i=0;i<A.size();i++){
if(i<a){
A[i]=0;
}else if(i-a<b){
A[i]=1;
}else{
A[i]=2;
}
}
}