三向切分快速排序(golang实现)

封装成函数：

//三向切分快速排序
func ThreeWayQuickSort(s []int) {
    sort3way(s, 0, len(s)-1)
}
 
//在lt之前的(lo~lt-1)都小于中间值
//在gt之前的(gt+1~hi)都大于中间值
//在lt~i-1的都等于中间值
//在i~gt的都还不确定（最终i会大于gt，即不确定的将不复存在）
func sort3way(s []int, lo, hi int) {
    if lo >= hi {
        return
    }
    v, lt, i, gt := s[lo], lo, lo+1, hi
    for i <= gt {
        if s[i] < v {
            swap(s, i, lt)
            lt++
            i++
        } else if s[i] > v {
            swap(s, i, gt)
            gt--
        } else {
            i++
        }
    }
    sort3way(s, lo, lt-1)
    sort3way(s, gt+1, hi)
}
 
func swap(s []int, i int, j int) {
    s[i], s[j] = s[j], s[i]
}

测试：

s := []int{9, 0, 6, 5, 8, 2, 1, 7, 4, 3}
fmt.Println(s)
ThreeWayQuickSort(s)
fmt.Println(s)

输出：
[9 0 6 5 8 2 1 7 4 3]
[0 1 2 3 4 5 6 7 8 9]

有疑问加站长微信联系（非本文作者）