使用一次遍历完成彩球排序问题

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

思想为利用两个指针， p0来交换0,p1 来交换1，利用 i 完成一次遍历求得正确结果
代码如下

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;
        int p0 = 0, p1 = 0;
        for (int i = 0; i < n; ++i) {
            if (nums[i] == 1) {
                int temp = nums[i];
                nums[i] = nums[p1];
                nums[p1] = temp;
                ++p1;
            } else if (nums[i] == 0) {
                int temp = nums[i];		//交换p0与当前位置i的值
                nums[i] = nums[p0];
                nums[p0] = temp;
                if (p0 < p1) {
                    temp = nums[i];
                    nums[i] = nums[p1];
                    nums[p1] = temp;
                }
                ++p0;
                ++p1;
            }
        }
    }
}

用指针 p0来交换0,p1 来交换1
如果遍历时num[i]找到了1，那么将其与p1进行交换，并将p1向后移动一个位置，此时不需考虑过多。
如果遍历时num[i]找到了0，那么将其与p0进行交换，此时需要注意，因为p0指向的是0的末尾的后一位（也就是1的首位，如图）

此时置换num[i]与p0的值，会出现如下结果

原有的1会被错误得置换到前面，导致这种情况的原因是p1指向的数值之前必为排好序的1。（换言之，p1若在p2之前，则一定会出现这种错误情况）解决方法则是在置换num[i]与p0的值之后，加上num[i]与p1的置换。（效果是被置换出来的尾部的1重新回到头部，仍在1的队列保持有序）如图：

进行此过程，直到 i 循环到尽头，完成遍历一次得到结果