75. 颜色分类（LeetCode)

题目条件

---

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例：

输入：nums = [2,0,2,1,1,0]

输出：[0,0,1,1,2,2]

条件分析

---

1. 数据元素有限，只有3种
2. 要求原地排序，不申请多余的空间，考虑使用交换

解题思路

---

1. 可以定义两个下标zero和two，其中[0,zero]中包含的都是0，(zero,two)之间包含的都是1，[tow,nums.length-1]包含都是2；
2. 在遍历元素时进行比较，如果元素=1，i++,如果元素=0，交换i和zero的位置，i++,如果元素==2,交换i和two的位置,此时i不需要++，因为还不知道two所在位置的元素是多少

编码如下

public void sortColors(int[] nums) {
        int zero = -1; // [0,zero] == 0
	int two = nums.length; // [two, nums.length-1] == 2
	int cur = 0;
	while (cur < two) {
	        if (nums[cur] == 1) {
			cur++;
		} else if (nums[cur] == 0) {
			zero++;
			NumberUtils.swap(nums, cur, zero);
			cur++;
		} else {
			two--;
			NumberUtils.swap(nums, cur, two);
		}
	}
}