[Golang]力扣Leetcode—中级算法—排序和搜索—颜色分类（指针）

[Golang]力扣Leetcode—中级算法—排序和搜索—颜色分类（指针）

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

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

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

链接： 力扣Leetcode—中级算法—排序和搜索—颜色分类.

示例 1:

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

示例 2:

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

标签：数组、双指针、排序

思路：利用三个指针 left 表示0的右边界，right 表示2的左边界，cur 表示当前遍历的位置。( left <= cur <= right )

1. 当nums[cur]==0时，与left互换，并且left++ cur++。因为left一定<=cur，所以换过来的数字肯定是0或者1，所以cur++。
2. 当nums[cur]==2时，与right互换，并且right–，因为right对应的数字没有扫描过，不确定换回来的数字的大小，因此cur不变，下次循环需要再遍历一次cur处的值。
3. cur<=right而不是小于right，因为right位置的数还没有扫描过

全部Go代码如下：

package main
 
import "fmt"
 
func main() {
	var nums = []int{2, 0, 2, 1, 1, 0}
	var n = len(nums)
	// 当前位置
	var cur int
	// 0 的右边界
	var left int
	// 2 的左边界
	var right = n - 1
 
	var tmp int
 
	for cur <= right {
		// 当nums[cur]==0时，与left互换，并且left++ cur++。因为left一定<=cur，所以换过来的数字肯定是0或者1，所以cur++。
		if nums[cur] == 0 {
			tmp = nums[left]
			nums[left] = nums[cur]
			nums[cur] = tmp
			left++
			cur++
		} else if nums[cur] == 2 {
			// 当nums[cur]==2时，与right互换，并且right--，因为right对应的数字没有扫描过，不确定换回来的数字的大小，因此cur不变，下次循环需要再遍历一次cur处的值。
			tmp = nums[right]
			nums[right] = nums[cur]
			nums[cur] = tmp
			right--
		} else {
			// cur<=right而不是小于right，因为right位置的数还没有扫描过
			cur++
		}
	}
	fmt.Println(nums)
}

提交截图：