75. 颜色分类

image.png

解法

class Solution {
    public void sortColors(int[] nums) {
        int left = 0;
        int right = nums.length - 1;
        for (int i = 0; i <= right; i++) {
            // 遇到为2的，交换到最后面right指针，指针减1
            // while是因为新交换过来可能还是2
            while(i < right && nums[i] == 2) {
                int temp = nums[i];
                nums[i] = nums[right];
                nums[right] = temp;
                right--;
            }
            // 遇到为0的，交换到最前面left指针，指针加1
            if (nums[i] == 0) {
                int temp = nums[i];
                nums[i] = nums[left];      
                nums[left] = temp;
                left++;
            }            
        }
    }
}