题目:给定一个蕴含红色、红色和蓝色、共 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 mainimport "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)}

提交截图