[力扣c++实现] 75. 颜色分类

75. 颜色分类

难度中等1099

给定一个包含红色、白色和蓝色，一共 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

进阶：

• 你可以不使用代码库中的排序函数来解决这道题吗？
• 你能想出一个仅使用常数空间的一趟扫描算法吗？

1.C++

/*
* 2021-12-6 21:16
* Hangzhou
*
*/

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int left = 0;
        int right = nums.size() -1;

        while (left < right)
        {
            while (left < right && nums[left] == 0)
            {
                ++left;
            }

            while (left < right && nums[right] == 2)
            {
                --right;
            }

            if (nums[left] > nums[right])
            {
                nums[left] ^= nums[right];
                nums[right] ^= nums[left];
                nums[left] ^= nums[right];
            }  
            else 
            {
                int r = right - 1;
                while (r > left)
                {
                    if (nums[r] > nums[right])
                    {
                        nums[r] ^= nums[right];
                        nums[right] ^= nums[r];
                        nums[r] ^= nums[right];
                        break;
                    }
                    --r;
                }

                int l = left + 1;
                while (l < right)
                {
                    if (nums[l] < nums[left])
                    {
                        nums[l] ^= nums[left];
                        nums[left] ^= nums[l];
                        nums[l] ^= nums[left];
                        break;
                    }
                    ++l;
                }
                ++left;
            }
        }
        
        return ;
    }
};

int main()
{
    Solution slu
    vector<num> nums = {2,0,2,1,1,0,2,2,2,2,1,1,1,1,0,1,2,2,1,0,1,2,1,1,1,1,0,0,0,1,1,1};
    slu.sortColors(nums);
    return 0;
}