耐心和持久胜过激烈和*。
哈喽大家好,我是陈明勇,今天分享的内容是使用 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
,待放最小值的位置,那么最大值就会先被换走,所以需要判断一下,纠正下标值。