使用 Go 实现快速排序

Posted 2021-05-02 Golang语言社区

快速排序(quick sort)号称是二十世纪最伟大的十大算法之一(The Best of the 20th Century: Editors Name Top 10 Algorithms), 但是快速排序也是最不容易实现的排序算法之一 。虽然它的原理非常的简单，但实现起来很容易出错。 也曾因为快排导致腥风血雨甚至网站攻击事件。

快速排序由C. A. R. Hoare在1962年提出。它的基本思想是：通过一趟排序将要排序的数据分割成独立的两部分，其中一部分的所有数据都比另外一部分的所有数据都要小，然后再按此方法对这两部分数据分别进行快速排序，整个排序过程可以递归进行，以此达到整个数据变成有序序列。

分治法：将问题分解为若干个规模更小但结构与原问题相似的子问题。递归地解这些子问题，然后将这些子问题的解组合为原问题的解。

利用分治法可将快速排序的分为三步：

• 在数据集之中，选择一个元素作为”基准”（pivot）。

• 所有小于”基准”的元素，都移到”基准”的左边；所有大于”基准”的元素，都移到”基准”的右边。这个操作称为分区 (partition) 操作，分区操作结束后，基准元素所处的位置就是最终排序后它的位置。

• 对”基准”左边和右边的两个子集，不断重复第一步和第二步，直到所有子集只剩下一个元素为止。

快速排序平均时间复杂度为O(n log n),最坏情况为O(n2)，不稳定排序。

快速排序一般实现为原地排序(in-place),因为非原地排序会设计到大量的容器创建和对象复制。

本文实现了两种快速排序，一种是单线程的快速排序,一种是一定数量的goroutine并行的快速排序。

同时也增加了标准库排序算法和timsort算法的比较。

下面是算法实现：

 1package main
 2import (
 3    "fmt"
 4    "math/rand"
 5    "sort"
 6    "time"
 7    "github.com/psilva261/timsort"
 8)
 9func partition(a []int, lo, hi int) int {
10    pivot := a[hi]
11    i := lo - 1
12    for j := lo; j < hi; j++ {
13        if a[j] < pivot {
14            i++
15            a[j], a[i] = a[i], a[j]
16        }
17    }
18    a[i+1], a[hi] = a[hi], a[i+1]
19    return i + 1
20}
21func quickSort(a []int, lo, hi int) {
22    if lo >= hi {
23        return
24    }
25    p := partition(a, lo, hi)
26    quickSort(a, lo, p-1)
27    quickSort(a, p+1, hi)
28}
29func quickSort_go(a []int, lo, hi int, done chan struct{}, depth int) {
30    if lo >= hi {
31        done <- struct{}{}
32        return
33    }
34    depth--
35    p := partition(a, lo, hi)
36    if depth > 0 {
37        childDone := make(chan struct{}, 2)
38        go quickSort_go(a, lo, p-1, childDone, depth)
39        go quickSort_go(a, p+1, hi, childDone, depth)
40        <-childDone
41        <-childDone
42    } else {
43        quickSort(a, lo, p-1)
44        quickSort(a, p+1, hi)
45    }
46    done <- struct{}{}
47}
48func main() {
49    rand.Seed(time.Now().UnixNano())
50    testData1, testData2, testData3, testData4 := make([]int, 0, 100000000), make([]int, 0, 100000000), make([]int, 0, 100000000), make([]int, 0, 100000000)
51    times := 100000000
52    for i := 0; i < times; i++ {
53        val := rand.Intn(20000000)
54        testData1 = append(testData1, val)
55        testData2 = append(testData2, val)
56        testData3 = append(testData3, val)
57        testData4 = append(testData4, val)
58    }
59    start := time.Now()
60    quickSort(testData1, 0, len(testData1)-1)
61    fmt.Println("single goroutine: ", time.Now().Sub(start))
62    if !sort.IntsAreSorted(testData1) {
63        fmt.Println("wrong quick_sort implementation")
64    }
65    done := make(chan struct{})
66    start = time.Now()
67    go quickSort_go(testData2, 0, len(testData2)-1, done, 5)
68    <-done
69    fmt.Println("multiple goroutine: ", time.Now().Sub(start))
70    if !sort.IntsAreSorted(testData2) {
71        fmt.Println("wrong quickSort_go implementation")
72    }
73    start = time.Now()
74    sort.Ints(testData3)
75    fmt.Println("std lib: ", time.Now().Sub(start))
76    if !sort.IntsAreSorted(testData3) {
77        fmt.Println("wrong std lib implementation")
78    }
79    start = time.Now()
80    timsort.Ints(testData4, func(a, b int) bool { return a <= b })
81    fmt.Println("timsort: ", time.Now().Sub(start))
82    if !sort.IntsAreSorted(testData4) {
83        fmt.Println("wrong timsort implementation")
84    }
85}

---

版权申明：内容来源网络，版权归原创者所有。除非无法确认，我们都会标明作者及出处，如有侵权烦请告知，我们会立即删除并表示歉意。谢谢。

Golang语言社区

ID：Golangweb

游戏服务器架构丨分布式技术丨大数据丨游戏算法学习