Number.75——颜色分类

题目链接：https://leetcode-cn.com/problems/sort-colors/

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

解题过程

方法一：计数排序，两次扫描

先统计出每个颜色的个数，然后再向数组中填充对应个数的颜色。

class Solution {
    public void sortColors(int[] nums) {
        int count_0 = 0, count_1 = 0, count_2 = 0;
        for (int num : nums){
            if (num == 0)   count_0++;
            else if (num == 1) count_1++;
            else count_2++;
        }
        int i = 0;
        while (i < nums.length){
            if (i < count_0)    nums[i++] = 0;
            else if (i < count_0 + count_1)     nums[i++] = 1;
            else nums[i++] = 2;
        }
    }
}

方法二：一次扫描，常熟空间

定义三个指针：

• left：左边0的右边界
• right：右边2的左边界
• cur：当前遍历元素

算法过程：

• 当nums[cur]=0时，与nums[left]交换位置，并且移动left即0的右边界和移动cur，因为前面元素是0了
• 当nums[cur]=2时，与nums[right]交换位置，并且移动right即2的左边界，但是不移动cur，因为交换后的nums[cur]可能不为0
• 当nums[cur]=1时，cur加一，其它不操作

这样最后就会形成题目要求的顺序。

class Solution {
    public void sortColors(int[] nums) {
        int left = 0;   // 左边0的右边界
        int right = nums.length - 1;    // 右边2的左边界
        int cur = 0;    // 当前元素

        while (cur <= right){
            if (nums[cur] == 0) {
                int temp = nums[cur];
                nums[cur] = nums[left];
                nums[left] = temp;
                left++;
                cur++;
            }else if (nums[cur] == 2){
                int temp = nums[cur];
                nums[cur] = nums[right];
                nums[right] = temp;
                right--;
            }else{
                cur++;
            }
        }
    }
}