75. Sort Colors（颜色分类）解法（C++ & 注释）

1. 题目描述

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

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

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

示例:

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

进阶：

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

题目链接：中文题目；英文题目

2. 指针法（Pointers）

2.1 解题思路

我们先把这题简化一下：如果数组里面只有两个颜色（red和blue）；那思路也比较简单，设置左边界和右边界两个指针：

1. nums[left] == red, left++;
2. nums[left] == blue,交换nums[right]和nums[left]，left++，right–;

其实这个思路也可以这么理解：我们右边界的指针表示blue所在的位置，如果左指针遇到blue，那么需要把blue放在右指针所在的位置，交换后自然而然左指针的位置就变成了red，所以之后左指针右移一位，右指针左移一位。同样，左指针表示red所在的位置。

现在我们现在把颜色增加一个（red, white, blue），根据上面的思路，我们同样设置左指针（redIdx）和右指针（blueIdx），那么现在还差一个white的指针（whiteIdx），我们把white的指针放在和左指针一样的位置。

1. nums[whiteIdx] == red, 交换nums[whiteIdx]和nums[redIdx]，whiteIdx++，redIdx++；
2. nums[whiteIdx] == white，whiteIdx++；
3. nums[whiteIdx] == blue，交换nums[whiteIdx]和nums[blue]，blueIdx–；

总之，我们用指针表示每个颜色所在的正确位置，以其中一个颜色为参考，从左或从右移动这个参考颜色，遇到其他颜色，两两交换；如果遇到自己的颜色，则继续往后检查。注意，这里是可以whiteIdx = blueIdx，因为redIdx还可能和whiteIdx进行交换。

最后，题目说道的Tow-pass代码，大家可以参考：Four different solutions第一个代码 - two pass O(m+n) space；

2.2 实例代码

typedef enum colors {RED, WHITE, BLUE};

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int redIdx = 0, whiteIdx = 0, blueIdx = nums.size() - 1;

        while (whiteIdx <= blueIdx) {
            if (nums[whiteIdx] == BLUE) { swap(nums[whiteIdx], nums[blueIdx]); blueIdx--; }
            else if (nums[whiteIdx] == RED) { swap(nums[whiteIdx], nums[redIdx]); whiteIdx++; redIdx++; }
            else whiteIdx++;
        }
    }
};

3. 参考资料

1. Python O(n) 1 pass in-place solution with explanation
2. Dutch national flag problem
3. Four different solutions

---