Sort Colors

LeetCode Short Colour Array problem

// JAVA CODE

class Solution {
    public void sortColors(int[] nums) {
        int zero=0,one=0,two=0;
        for(int i:nums){
            if(i==0) zero++;
            if(i==1) one++;
            if(i==2) two++;
        }
        int i=0;
        while(zero-->0)
            nums[i++]=0;
        while(one-->0)
            nums[i++]=1;
        while(two-->0)
            nums[i++]=2;
        
    }
}


//////

INPUT
Input: nums = [2,0,2,1,1,0]
/////

OUTPUT
Output: [0,0,1,1,2,2]