75. 颜色分类

题目

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

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

必须在不使用库的sort函数的情况下解决这个问题。

示例 1：

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

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

提示：

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

链接

75. 颜色分类

快慢指针，遍历两次

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

荷兰国旗问题，三指针

class Solution {
public:
    void sortColors(vector<int>& nums) {
        // 荷兰国旗问题：三个指针进行记录
        // i、j指针从0开始，k指针从末尾开始
        // i指针指向当前遍历的数字 j指针指向需要存放0的位置 k指针指向需要存放2的位置
        // 0~j-1全为0   j~i-1全为1   i~k-1为需要遍历判断的数字  k~nums.size()-1全为2
        for(int i = 0, j = 0, k = nums.size()-1; i <= k; ){
            if(nums[i] == 0) swap(nums[i++], nums[j++]);//如果当前数字是0，则i位置跟j位置元素交换，并都后移一个位置，因为i跟j可以看成是快慢指针，i指针的作用确保了j位置的元素只有0或者1
            else if(nums[i] == 2) swap(nums[i], nums[k--]);//如果当前数字是2,则跟把k位置的0或者1换到当前位置i，下次循环继续判断i，进行交换
            else if(nums[i] == 1) i++;//如果是1,则跳过，继续判断下一个数字
        }
    }
};