Go 实现选择排序算法

时间:2022-12-09 07:20:41

耐心和持久胜过激烈和*。

哈喽大家好,我是陈明勇,今天分享的内容是使用 Go 实现选择排序算法。如果本文对你有帮助,不妨点个赞,如果你是 Go 语言初学者,不妨点个关注,一起成长一起进步,如果本文有错误的地方,欢迎指出!

选择排序

选择排序是一种简单的比较排序算法,它的算法思路是首先从数组中寻找最小(大)的元素,然后放到数组中的第一位,接下来继续从未排序的数组中寻找最小(大)元素,然后放到已排序数组的末尾,依次类推,直到所有元素被排序。

图片演示

普通算法

import "fmt"

func main() {
nums := [4]int{3, 1, 4, 2}
fmt.Println("原数组:", nums)
fmt.Println("--------------------------------")
SelectionSort(nums)
}

func SelectionSort(nums [4]int) {
for i := 0; i < len(nums)-1; i++ {
minPos := i
for j := i + 1; j < len(nums); j++ {
if nums[minPos] > nums[j] {
minPos = j
}
}
nums[i], nums[minPos] = nums[minPos], nums[i]
fmt.Printf("第 %d 轮后:%v\n", i+1, nums)
}
fmt.Println("--------------------------------")
fmt.Println("排序后的数组:", nums)
}

执行结果:

原数组: [3 1 4 2]
--------------------------------
第 1 轮后:[1 3 4 2]
第 2 轮后:[1 2 4 3]
第 3 轮后:[1 2 3 4]
--------------------------------
排序后的数组: [1 2 3 4]

升序排序,使用 ​​i​​ 变量表示最小元素的待放位置, ​​minPos​​ 变量记录最小元素的的下标值,默认为 ​​i​​,然后通过变量 ​​j​​ 去寻找最小元素,​​j​​ 从 ​​i + 1​​ 的位置开始寻找,找到比 ​​nums[minPos]​​ 还小的元素,则将 ​​j​​ 的下标值赋给 ​​minPos​​,一轮下来,将最小元素的位置 ​​minPos​​ 与 ​​i​​ 的位置互换,然后继续下一轮寻找,直到所有元素都被排序。 该算法的时间复杂度为 O(N²)。

优化算法

普通算法是寻找最小值/最大值,然后放到指定位置,既然是寻找值,我们可以同时最小值和最大值

import (
"fmt"
)

func main() {
nums := [4]int{3, 1, 4, 2}
fmt.Println("原数组:", nums)
fmt.Println("--------------------------------")
SelectionSort(nums)
}

func SelectionSort(nums [4]int) {
for left, right := 0, len(nums)-1; left <= right; {
minPos := left
maxPos := left
for i := left + 1; i <= right; i++ {
if nums[minPos] > nums[i] {
minPos = i
}
if nums[maxPos] < nums[i] {
maxPos = i
}
}
nums[left], nums[minPos] = nums[minPos], nums[left]
// 如果最大值刚好是在 left,待放最小值的位置,那么最大值就会被换走,所以需要判断一下
if maxPos == left {
maxPos = minPos
}
nums[right], nums[maxPos] = nums[maxPos], nums[right]
fmt.Printf("第 %d 轮后:%v\n", left+1, nums)
left++
right--
}
fmt.Println("--------------------------------")
fmt.Println("排序后的数组:", nums)
}

有一个注意的地方是,如果最大值刚好是在 ​​left​​ ,待放最小值的位置,那么最大值就会先被换走,所以需要判断一下,纠正下标值。