leetcode75. Sort Colors

75. Sort Colors

帅气

class Solution {
    public void sortColors(int[] nums) {
        int zero =0;
        int two = nums.length -1;
        int i =0;
        while(i<nums.length){
            while(zero < nums.length && nums[zero]==0){
                zero++;
            }
            while(two >= 0  && nums[two]==2){
                two--;
            }
            if(nums[i]==0 && i> zero){
                swap(nums,i,zero);
                zero++;
            }
            else if(nums[i]==2 && i<two){
                swap(nums,i,two);
                two--;
            }
            else{
                i++;
            }
            
        }
      
    }
    private void swap(int[] nums,int i,int j){
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}