算法 in Golang:Quicksort(快速排序)

发布时间 2023-06-06 22:10:03作者: 寻月隐君

算法 in Golang:Quicksort(快速排序)

Quicksort(快速排序)

  • 快速排序 O(nlog2^n),比选择排序要快 O(n²)
  • 在日常生活中经常使用
  • 使用了 D & C 策略(分而治之)

使用 Quicksort 排序数组

  • 不需要排序的数组(也就是 Base Case 基线条件):
    • [],空数组
    • [s],单元素数组
  • 很容易排序的数组:
    • [a, b],两个元素的数组,只需检查它们之间的大小即可,调换位置
  • 3 个元素的数组(例如 [23, 19, 35]):
    • 使用 D & C 策略,简化至基线条件(Base case)
  1. 从数组中随便选一个元素,例如 35,这个元素叫做 pivot(基准元素)

  2. 找到比 pivot 小的元素,找到比 pivot 大的元素,这叫做分区:[23, 19], (35), []

  3. 如果左右两个子数组已排好序(达到基线条件),结果:左边 + [pivot] + 右边

  4. 如果左右两个子数组没排好序(没达到基线条件),那么:

    quicksort(左边) + [pivot] + quicksort(右边)

使用 Quicksort 排序数组的步骤

  1. 选择一个 pivot
  2. 将数组分为两个子数组:
    1. 左侧数组的元素都比 pivot 小
    2. 右侧数组的元素都比 pivot 大
  3. 在两个子数组上递归的调用 quicksort

创建项目

~/Code/go via ? v1.20.3 via ? base
➜ mcd quicksort

Code/go/quicksort via ? v1.20.3 via ? base
➜ go mod init quicksort
go: creating new go.mod: module quicksort

Code/go/quicksort via ? v1.20.3 via ? base
➜ c

Code/go/quicksort via ? v1.20.3 via ? base
➜

main.go 代码

package main

import "fmt"

func main() {
	arr := []int{12, 87, 1, 66, 30, 126, 328, 12, 653, 67, 98, 3, 256, 5, 1, 1, 99, 109, 17, 70, 4}
	result := quicksort(arr)
	fmt.Println("result: ", result)
}

func quicksort(arr []int) []int {
	if len(arr) < 2 {
		return arr
	}

	pivot := arr[0]
	var left, right []int

	for _, ele := range arr[1:] {
		if ele <= pivot {
			left = append(left, ele)
		} else {
			right = append(right, ele)
		}
	}
	return append(quicksort(left), append([]int{pivot}, quicksort(right)...)...)
}

运行

Code/go/quicksort via ? v1.20.3 via ? base 
➜ go run main.go       
result:  [1 1 1 3 4 5 12 12 17 30 66 67 70 87 98 99 109 126 256 328 653]

Code/go/quicksort via ? v1.20.3 via ? base took 3.2s 
➜