sortcolors.cpp

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int arr[] ={0,0,0};
        for(int i = 0; i < nums.size(); i++)
            arr[nums[i]]++;
        int next = 0;
        for(int i = 0; i < nums.size(); i++){
            while(arr[next] > 0){
                nums[i] = next;
                arr[next]--;
                i++;
            }
            next++;
            i--;
        }
    }
};