[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)
}

提交截图：