leetcode75 颜色分类

75 颜色分类
给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。
示例
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶：
一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-colors
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路
直接来考虑进阶的问题，要想实现仅用常数空间的一趟扫描算法，因为题目中的颜色事实上只有三种，分别代表为0、1、2，最容易考虑到的是采用双指针a1、a2从数组nums的两侧分别进行扫描，再利用另外一个指针m顺序遍历比较，具体如下：
1、初始化指针位置
2、当nums[m]=2时，交换与nums[a2]的位置，这种情况m位置不能改变，因为可能存在二者都是2的情况，m++会使得这种情况被跳过
3、当nuns[m[=1时，m直接加1，它并不需要改变
4、当nums[m]=0时，，交换与nums[a1]的位置，这种情况m可以直接加一，因为a1所指代的是0，本就应该在分类后的数组的最前面

代码

func sortColors(nums []int)  {
    var length int = len(nums)
	a1 := 0
	a2 := length - 1
	m := 0
	  
	for m <= a2{
	   if nums[m] == 2{
		   	   nums[m] = nums[a2]
			   nums[a2] = 2
			   a2--
		    }else if nums[m] == 1{
		   	   m++
		    }else if nums[m] == 0{
		   	   nums[m] = nums[a1]
			   nums[a1] = 0
			   a1++
			   m++
		    }
	    }
}

运行结果示例
调用

package main
import "fmt"

func main(){
       nums := []int{2,0,2,1,1,0}
	   //var a1,a2,m int 
	   sortColors(nums)
       fmt.Println("%d",nums)
}

结果

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