荷兰国旗问题——力扣75

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶：

• 一个直观的解决方案是使用计数排序的两趟扫描算法。
  首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
• 你能想出一个仅使用常数空间的一趟扫描算法吗？

方法一：原地排序 

package Sorts;

public class SolutionSortColors {
    public static void sortColors(int[] nums) {
        //使用one进行遍历，zero代表one的前一个，two控制循环的结束
        int zero = -1, one = 0, two = nums.length;
        while (one < two) {
            if (nums[one] == 0) {
                /**如果当前数组的值为0，那么下面这行代码的意思是，
                 * 将这个数值0的位置和它的前一个数值调换，目的是使数值为0的颜色始终保持在前面
                 * 将zero和one都向后移动
                 * */
                swap(nums, ++zero, one++);
            } else if (nums[one] == 2) {
                /**如果当前数组的值为2，那么下面这行代码的意思是，
                 * 将这个数值2的位置和数组最后面的元素调换位置，目的是使数值为2的颜色始终保持在后面
                 * two向前移动，交换后one的位置不动，继续进行操作，因为不确定交换后one的位置的数值是多少
                 * */
                swap(nums, --two, one);
            } else {
                /**只剩下数值为1的情况，直接跳过不进行操作
                 * */
                one++;
            }
        }
    }

    public static void swap(int[] nums, int i, int j) {
        //交换前后两个数值
        int t = nums[i];
        nums[i] = nums[j];
        nums[j] = t;
    }
    public static void main(String[] args) {
        int[] nums = new int[]{2, 0, 2, 1, 1, 0};
        sortColors(nums);
        for (int num : nums) {
            System.out.println(num);
        }
    }
}