75. Sort Colors

SortColors.java

public class SortColors {
    public void sortColors(int[] nums) {
        if(nums.length > 1) {
            int redIndex = 0;
            int blueIndex = nums.length-1;
            int i = 0;
            while(i <= blueIndex) {
                if(nums[i] == 2) {
                    // Send blue node to end of array
                    swap(nums, i, blueIndex--);
                    // Do not proceed to next node, need to check new value for red
                } else if(nums[i] == 0) {
                    // Send red to beginning of array
                    swap(nums, i, redIndex++);
                    i++;
                } else {
                    i++;
                }
            }
        }
    }

    private void swap(int[] nums, int index, int swapIndex) {
        int temp = nums[swapIndex];
        nums[swapIndex] = nums[index];
        nums[index] = temp;
    }
}