75 sort color

class Solution {
    public void sortColors(int[] nums) {
        int r = 0;
        int b = nums.length -1;
        int i =0;
        while(i <= b){
            if(nums[i] == 0){
                nums[i] = 1;
                nums[r] = 0;
                r++;
                i++;
            }else if(nums[i] == 1){
                i++;
            }else{
                nums[i] = nums[b];
                nums[b] = 2;
                b--;
            } 
        }
    }
}