给定一个包含红色、白色和蓝色，一共 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]
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-colors
解题思路
用两个指针, 一左一右zero two, 分别用来将 0 和 2 放到前面和后面
遍历数组, 循环条件 i <= two, 因为遍历时从左往右, 将 2 归位从右往左, 当遍历的位置超过two时就要结束
如果num[i] = 2, 交换i和two--所指的元素, --是更新two的位置, 往前挪
由于换位置后可能num[i]还是 2, 所以要继续交换i和two--所指的元素直到i所指元素不为2
再判断是否i所指为 0, 如果是则交换i和zero++所指的元素
代码
class Solution {
public void sortColors(int[] nums) {
int zero = 0; // 用于将0换到最前面
int two = nums.length - 1; // 用于将2放到最后面
for (int i = 0; i <= two; i++) {
while (i <= two && nums[i] == 2) {
swap(nums, two--, i);
}
if (nums[i] == 0) {
swap(nums, zero++, i);
}
}
}
public void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}
