75 颜色分类

题目

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

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

必须在不使用库内置的 sort 函数的情况下解决这个问题。

示例

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

解析

这道题，主要的思路是双指针，但是和之前的双指针意思不一样，这次因为数组中有三种元素，用zero指针和two指针分别代表0和2，然后剩下的中间的就是1.

func sortColors(nums []int) {
	n := len(nums)
	zero, two := 0, n-1
	for i := 0; i < two; { // two初始化在末尾，然后不断被更新，遍历到two的时候其实后面都是确认过的2了
		if nums[i] == 1 { // 1什么都不用做
			i++
		} else if nums[i] == 2 { 
			nums[i], nums[two] = nums[two], nums[i]
			two -= 1
		} else {
			nums[i], nums[zero] = nums[zero], nums[i]
			zero += 1
			i++
		}
	}
}

上面的代码主要涉及的地方其实就是等于2的时候，直接交换了并没有i++，其实是这样：最开始zero指向的元素可能并不是0，two指向的元素也并不是2，比如首次和2交换后，交换过去的元素可能012都有可能，那么如果直接i++的话就出错了；但是和0交换过去后，zero也更新了，需要继续走下去才能继续遍历。
实在想不明白的话这题画个图模拟一下吧，不是很好想