一、题目
给定一个包含红色、白色和蓝色，一共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
二、解题思路
数组中仅包含三种数字：0、1、2。我们只需将 0 移至前面，将 3 移至后面即可。难度并非很大。下面以示例一为例，来看如下动图。
三、代码实现
#include <iostream>
#include <vector>using namespace std;// 交换数组中的两个数字
void swap(vector<int>& nums, int i, int j) {int tmp = nums[i];nums[i] = nums[j];nums[j] = tmp;
}// 三指针排序
void sortColors(vector<int>& nums) {// 0的右边界int left = 0;// 2的左边界int right = nums.size() - 1;// 指向当前数字int index = 0;while (index <= right) {if (nums[index] == 0) {// 如果是0，就往前面移swap(nums, left++, index++);} else if (nums[index] == 1) {index++;} else if (nums[index] == 2) {// 如果是2就往后面移swap(nums, right--, index);}}
}int main() {vector<int> nums = {2, 0, 2, 1, 1, 0};sortColors(nums);for (int num : nums) {cout << num << " ";}cout << endl;return 0;
}
