leetcode之75颜色分类Golang

题目描述

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

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

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

示例:

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

进阶：

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

算法描述

定义左中右三个指针，他们分别代表的值是0，1，2，三个指针的初始值是0，0，len

中间指针遍历数组，当中间指针的值是1，就往后遍历，当中间指针的指向的值是0，就和左边指针的值交换并且向左移左边指针，当中间指针的值是2，就和右边指针的值交换，并且向左移动右边指针

当中间指针和右边指针相遇，算法结束

代码

func sortColors(nums []int) {
	low, mid, high := 0, 0, len(nums)-1
	// 定义三个指针，当中间的指针超过右边的指针时，就返回了
	// 算法核心：
	for mid <= high {
		if nums[mid] == 1 {
			// 当中间指针指向的值是1时，中间指针向后移动一位
			mid++
			continue
		} else if nums[mid] == 0 {
			// 交换中间指针和左边指针的值
			// 当中间指针指向的值是0时，如果此时左边指针和中间指针重合，那么两个指针都向后移动一位
			// 否则只是左边的指针向后移动一位
			nums[low], nums[mid] = nums[mid], nums[low]
			if mid == low {
				mid++
			}
			low++
		} else if nums[mid] == 2 {
			// 交换中间指针和右边指针的值
			// 当中间的指针指向的值是2时，右边的指针向左移动一位
			nums[high], nums[mid] = nums[mid], nums[high]
			high--
		}
	}
}