LeetCode 75. 颜色分类(C++)

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

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

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

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

进阶：

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

思路一（计数排序）：

统计各个颜色数量（0、1、2的数量），按照数量遍历数组进行赋值。

       if(nums.empty())
            return;
        int count[3] = {0};
        for(int i = 0; i < nums.size(); ++i){
            ++count[nums[i]];
        }
        for(int i = 0; i < count[0]; i++){
            nums[i] = 0;
        }
        for(int i = 0; i < count[1]; i++){
            nums[i + count[0]] = 1;
        }
         for(int i = 0; i < count[2]; i++){
            nums[i + count[1] + count[0]] =2;
        }

思路二（三路快排）：

设置0与1的边界为zero（初始值设置为-1，保证界内无元素），1与2的边界为two（初始值为nums.size(),保证界内无元素），遍历数组（索引为 i）
（1）若nums[ i ] == 1,i++
（2）若nums[ i ] == 0，则将zero右移一位，并将nums[ i ] 与 nums [ zero ]元素互换，i++
（3）nums[ i ] == 2的左移two，并将nums[ i ] 与 nums[ zero ] 互换，注意，此时 i 不变。

        if(nums.empty())
            return;
        int zero = -1;          //[0,zero]元素为1
        int two = nums.size();  //[two,nums.size() - 1]元素为2

        for(int i = 0; i < two; ){
            if(nums[i] == 1)
                ++i;
            else if(nums[i] == 0)
                swap(nums[++zero],nums[i++]);
            else if(nums[i] == 2)
                swap(nums[--two],nums[i]);
        }