75. 颜色分类

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

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

示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

提示：

n == nums.length
1 <= n <= 300
nums[i] 为 0、1 或 2

正确答案

class Solution {
public:
    void sortColors(vector<int>& nums) {
        sort(nums.begin(),nums.end());
    }
};

没想到那么简单。。。但是我看了一下答案

荷兰国旗问题

思路:借助快速排序partition过程的一趟扫描法
回顾快速排序partition过程︰随机选择一个元素作为切分元素(pivot)，然后经过一次扫描，通过交换不同位置的元素使得数组按照数值大小分成以下3个部分:

第一种解题方式（双指针+两头）

定义分区
all in [0, p0) == 0
all in [p0, i) == 1
all in (p2, len - 1] == 2
定义循环终止条件
i>p2

正确答案

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int len = nums.size();
        if (len < 2)
            return;
        //[0,p0)=0
        //[p0,i)=1
        //(p2,len-1)=2
        int p0 = 0;
        int i = 0;
        int p2 = len - 1;
        while (i <= p2){
            if (nums[i] == 0){
                swap(nums[i], nums[p0]);
                p0++; i++;
            }
            else if (nums[i] == 1)
                i++;
            else{
                swap(nums[i], nums[p2]);
                p2--;
            }
        }
    }
};

else中不需要向前移动i的原因：当我们将nums[i]与 nums[p2] 进行交换之后，新的nums[i]可能仍然是 2，也可能是 0。然而此时我们已经结束了交换，开始遍历下一个元素nums[i+1]，不会再考虑nums[i] 了，这样我们就会得到错误的答案。

第二种解题方式（双指针+两头）

定义分区
all in [0, p0] == 0
all in (p0, i) == 1
all in [p2, len - 1] == 2
定义循环终止条件
i>=p2

正确答案

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int len = nums.size();
        if (len < 2)
            return;
        //all in [0, p0] == 0
	    //all in (p0, i) == 1
	    //all in [p2, len - 1] == 2
        int p0 = -1;
        int i = 0;
        int p2 = len;
        while (i < p2){
            if (nums[i] == 0){
                p0++;
                swap(nums[i], nums[p0]);
                i++;
            }
            else if (nums[i] == 1)
                i++;
            else{
                p2--;
                swap(nums[i], nums[p2]);
            }
        }
    }
};

第三种解法（双指针+单头）

正确答案

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int n = nums.size();
        int p0 = 0, p1 = 0;
        for (int i = 0; i < n; ++i) {
            if (nums[i] == 1) {
                swap(nums[i], nums[p1]);
                ++p1;
            } else if (nums[i] == 0) {
                swap(nums[i], nums[p0]);
                if (p0 < p1) {
                    swap(nums[i], nums[p1]);
                }
                ++p0;
                ++p1;
            }
        }
    }
};

第四种解放（单指针）

正确答案

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int n = nums.size();
        int ptr = 0;
        for (int i = 0; i < n; ++i) {
            if (nums[i] == 0) {
                swap(nums[i], nums[ptr]);
                ++ptr;
            }
        }
        for (int i = ptr; i < n; ++i) {
            if (nums[i] == 1) {
                swap(nums[i], nums[ptr]);
                ++ptr;
            }
        }
    }
};