Leetcode——颜色分类（双指针，桶排序）

1. 颜色分类

（1）双指针（普通）

• 使用双指针left，right，指针left左侧都是0，指针right右侧都是2，初始化left，right在数组的两端。
• 使用一个遍历变量i，遍历数组时nums[i]为0，则与左指针指向数字交换位置，然后左指针和i都向后移动，i左侧直到左指针都是1，当nums[i]为2时，则与右指针指向数字交换位置，然后右指针向左移动，i不变因为还要判断交换来的数字是什么，如果nums[i]为1，则不交换i向后移动，循环终止条件时i超过了右指针，此时后面已经是2，无需继续交换

class Solution {
    public void sortColors(int[] nums) {
        int left = 0, right = nums.length - 1;
        int i = 0;
        while (i <= right) {
            //遍历数组时nums[i]为0，则与左指针指向数字交换位置，然后左指针和i都向后移动
            if (nums[i] == 0) {
                swap(nums, left, i);
                left++;
                i++;
            
            //当nums[i]为2时，则与右指针指向数字交换位置，然后右指针向左移动，i不变
            //为什么i不变，为了继续判断交换后的i是否为0，如为0还需要与左指针交换
            } else if (nums[i] == 2) {
                swap(nums, right, i);
                right--;
            
            //nums[i]为1，则不交换i向后移动，循环终止条件时i超过了右指针，此时后面已经是2，无需继续交换
            } else {
                i++;
            }
        }
    }

    public void swap(int [] nums, int i, int j) {
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }
}

（2）双指针（赋值）

• 无交换，简洁

class Solution {
    public void sortColors(int[] nums) {
        int left = 0, right = 0;
        for (int i = 0; i < nums.length; i++) {
            int num = nums[i];
            nums[i] = 2;        //最后剩下的值只有2

            if (num < 2) {        //只小于2，将left值赋为1,left+1 (碰见0时，原先的1会往后移一位)
                nums[left] = 1;
                left++;
            }               
            if (num < 1) {        //小于2也小于1，将right值赋为0，left和right都+1
                nums[right] = 0;
                right++;
            }
        }
    }
}

（3）桶排序

• 由于题目的值只有0，1，2，直接使用桶排序，统计0，1，2出现个数
• 然后再恢复即可，虽然桶排序不是一遍，但是相当于其他的快排，对于这道题来说是占优势的，快排为O(nlgn)；

class Solution {
    public void sortColors(int[] nums) {
        int[] bucket = new int[3];
        for (int num:nums) {
            bucket[num]++;
        }
        for (int i = 0, j = 0; i < 3; i++) {
            while(bucket[i] != 0){
                nums[j++] = i;
                bucket[i]--;
            }
        }
    }


}