关于golang:Golang力扣Leetcode中级算法排序和搜索颜色分类指针

10次阅读

共计 1050 个字符,预计需要花费 3 分钟才能阅读完成。

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

提交截图

正文完
 0