快排借助partition或荷兰国旗两种算法,将数组按既定的值进行分类。我们以升序为例,partition是在数组中将小于等于既定值的数放在数组的左边,同时要让小于等于既定值的数组最后一个值为既定值,将大于既定值的数放在数组的右边;荷兰国旗是将小于既定的值的数放在数组左边,等于既定值的数放在数组中间(可能有多个与既定值相同的值),大于既定值的数放在数组右边。这样一来,既定值所在的一个区域(partition)或既定值所在的一片区域(荷兰国旗)在数组中的位置就被固定了下来。我们只要操作既定值左边界之前的值及其既定值右边界后边的值,重复上述操作即可让整个数组有序。注意:左右边界外的值都是无序的。既定值的选取上,可以选择当前子数组中任意的一个值(下面实现中选择子数组的左边界),**经典的快排是在有效的数组范围内随机选择一个值作为既定值。在数学上,概率累加和使的该方式实现的快排的时间复杂度收敛于O(nlogn)。**下列的快排时间复杂度为O(N ^ 2),因为没有引入随机性。用partition和荷兰国旗算法实现的快排时间复杂度都是O(N^2)。
简单的实现一下:
func QuickSort(arr []int) {
if len(arr) <= 1 {
return
}
base, l, r :=arr[0], 0, len(arr)-1
for i := 1; i <= r; {
if arr[i] > base {
arr[i], arr[r] = arr[r], arr[i]
r--
continue
}
arr[i], arr[l] = arr[l], arr[i]
l, i = l + 1, i + 1
}
QuickSort(arr[:l])
QuickSort(arr[l+1:])
}
Benchmark测试
package main
import (
"math/rand"
"sort"
"testing"
)
var arr0 []int
var arr1 []int
func init() {
length := 1000000000
arr0 = make([]int, length)
for i := 0; i < length; i++ {
arr0[i] = rand.Intn(length)
}
arr1 = make([]int, length)
copy(arr1, arr0)
}
func BenchmarkQuickSort(b *testing.B) {
QuickSort(arr0)
}
func Benchmark(b *testing.B) {
sort.Ints(arr1)
}
结果还算可以,比sort包中的Ints()要快一倍左右
goos: windows
goarch: arm64
pkg: algorithem/quick_sort
BenchmarkQuickSort
BenchmarkQuickSort-8 1 119728626000 ns/op
Benchmark
Benchmark-8 1 242426651000 ns/op
PASS
看一下排序过程中的内存消耗