【算法六十一】75. 颜色分类

75. 颜色分类 - 力扣（LeetCode）

"骚"：

class Solution {
    public void sortColors(int[] nums) {
        //[0,p0-1]全是 0
        int p0 = 0;
        //[p0,p1-1]全是 1
        int p1 = 0;

        for(int i = 0;i<nums.length;i++){
            int x = nums[i];
            //无论 x 是几，2 都要先放后面
            nums[i] = 2;
            //x 是 0 和 1 时，1 都要换位置
            if(x<=1){
                nums[p1] = 1;
                p1++;
            }
            //x 是 0 时，0 要换位置
            if(x == 0){
                nums[p0] = 0;
                p0++;
            }
        }
    }
}

时间复杂度：O(N)

空间复杂度：O(1)

传统三指针：

class Solution {
    public void sortColors(int[] nums) {
        //0 的右边界（但不包含 left）
        int left = 0;
        //遍历到的位置
        int i = 0;
        //2 的左边界（但不包含 right）
        int right = nums.length - 1;

        while(i<=right){
            if(nums[i]==0){
                swap(nums,i,left);
                left++;
                i++;
            }
            else if(nums[i]==1){
                i++;
            }
            else{
                swap(nums,i,right);
                right--;
            }
        }
    }

    private void swap(int[] nums,int left,int right){
        int temp = nums[left];
        nums[left] = nums[right];
        nums[right] = temp;
    }
}

时间复杂度：O（N）

空间复杂度：O（1）